ci: update lmms-eval to speed up multimodal CI (#11000)

This commit is contained in:
b8zhong
2025-10-18 11:51:19 -07:00
committed by GitHub
parent c44e985dc2
commit f4f8a1b4d8
3 changed files with 13 additions and 5 deletions

View File

@@ -81,7 +81,7 @@ class TestVLMModels(CustomTestCase):
model = "openai_compatible"
tp = 1
tasks = "mmmu_val"
batch_size = 2
batch_size = 32
log_suffix = "openai_compatible"
os.makedirs(output_path, exist_ok=True)
@@ -179,7 +179,15 @@ class TestVLMModels(CustomTestCase):
self.run_mmmu_eval(model.model, output_path)
# Get the result file
result_file_path = glob.glob(f"{output_path}/*.json")[0]
# Search recursively for JSON result files (lmms-eval v0.4.1+ creates subdirectories)
result_files = glob.glob(f"{output_path}/**/*.json", recursive=True)
if not result_files:
result_files = glob.glob(f"{output_path}/*.json")
if not result_files:
raise FileNotFoundError(f"No JSON result files found in {output_path}")
result_file_path = result_files[0]
with open(result_file_path, "r") as f:
result = json.load(f)