Fix import warnings (#15144)
This commit is contained in:
@@ -36,7 +36,7 @@ def tensor_torch2ms(x: torch.Tensor):
|
||||
return ms_tensor
|
||||
|
||||
|
||||
def tensor_ms2torch(x: ms.Tensor):
|
||||
def tensor_ms2torch(x: "ms.Tensor"):
|
||||
if x is None or not isinstance(x, ms.Tensor):
|
||||
return x
|
||||
|
||||
|
||||
@@ -18,7 +18,7 @@ from sglang.test.test_utils import (
|
||||
MODEL_SCORE_THRESHOLDS = {
|
||||
"hugging-quants/Meta-Llama-3.1-8B-Instruct-AWQ-INT4": 0.825,
|
||||
"hugging-quants/Meta-Llama-3.1-8B-Instruct-GPTQ-INT4": 0.825,
|
||||
"hugging-quants/Mixtral-8x7B-Instruct-v0.1-AWQ-INT4": 0.62,
|
||||
"hugging-quants/Mixtral-8x7B-Instruct-v0.1-AWQ-INT4": 0.615,
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user