Avoiding the problem of printing a large number of compatibility warn… (#25956)
This commit is contained in:
@@ -42,6 +42,7 @@ class GSM8KAscendMixin(ABC):
|
|||||||
"SGLANG_ENBLE_TORCH_COMILE": "1",
|
"SGLANG_ENBLE_TORCH_COMILE": "1",
|
||||||
"AUTO_USE_UC_MEMORY": "0",
|
"AUTO_USE_UC_MEMORY": "0",
|
||||||
"P2P_HCCL_BUFFSIZE": "20",
|
"P2P_HCCL_BUFFSIZE": "20",
|
||||||
|
"TRANSFORMERS_VERBOSITY": os.getenv("TRANSFORMERS_VERBOSITY", "error"),
|
||||||
}
|
}
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|||||||
@@ -43,6 +43,10 @@ class TestVLMModels(CustomTestCase):
|
|||||||
os.environ["OPENAI_API_KEY"] = cls.api_key
|
os.environ["OPENAI_API_KEY"] = cls.api_key
|
||||||
os.environ["OPENAI_API_BASE"] = f"{cls.base_url}/v1"
|
os.environ["OPENAI_API_BASE"] = f"{cls.base_url}/v1"
|
||||||
|
|
||||||
|
os.environ["TRANSFORMERS_VERBOSITY"] = os.getenv(
|
||||||
|
"TRANSFORMERS_VERBOSITY", "error"
|
||||||
|
)
|
||||||
|
|
||||||
def run_mmmu_eval(
|
def run_mmmu_eval(
|
||||||
self,
|
self,
|
||||||
model_version: str,
|
model_version: str,
|
||||||
|
|||||||
Reference in New Issue
Block a user