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",
|
||||
"AUTO_USE_UC_MEMORY": "0",
|
||||
"P2P_HCCL_BUFFSIZE": "20",
|
||||
"TRANSFORMERS_VERBOSITY": os.getenv("TRANSFORMERS_VERBOSITY", "error"),
|
||||
}
|
||||
|
||||
@classmethod
|
||||
|
||||
@@ -43,6 +43,10 @@ class TestVLMModels(CustomTestCase):
|
||||
os.environ["OPENAI_API_KEY"] = cls.api_key
|
||||
os.environ["OPENAI_API_BASE"] = f"{cls.base_url}/v1"
|
||||
|
||||
os.environ["TRANSFORMERS_VERBOSITY"] = os.getenv(
|
||||
"TRANSFORMERS_VERBOSITY", "error"
|
||||
)
|
||||
|
||||
def run_mmmu_eval(
|
||||
self,
|
||||
model_version: str,
|
||||
|
||||
Reference in New Issue
Block a user