[CI] Tiny speed up VLM CI (#14517)
Co-authored-by: Brayden Zhong <b8zhong@users.noreply.github.com>
This commit is contained in:
@@ -14,7 +14,7 @@ if _is_hip:
|
||||
MODELS = [SimpleNamespace(model="openbmb/MiniCPM-V-2_6", mmmu_accuracy=0.4)]
|
||||
else:
|
||||
MODELS = [
|
||||
SimpleNamespace(model="google/gemma-3-27b-it", mmmu_accuracy=0.45),
|
||||
SimpleNamespace(model="google/gemma-3-4b-it", mmmu_accuracy=0.38),
|
||||
SimpleNamespace(model="Qwen/Qwen2.5-VL-3B-Instruct", mmmu_accuracy=0.4),
|
||||
SimpleNamespace(model="openbmb/MiniCPM-V-2_6", mmmu_accuracy=0.4),
|
||||
]
|
||||
|
||||
@@ -29,7 +29,7 @@ suites = {
|
||||
TestFile("models/test_qwen_models.py", 90),
|
||||
TestFile("models/test_reward_models.py", 103),
|
||||
TestFile("models/test_transformers_models.py", 245),
|
||||
TestFile("models/test_vlm_models.py", 282),
|
||||
TestFile("models/test_vlm_models.py", 270),
|
||||
TestFile("openai_server/basic/test_openai_embedding.py", 70),
|
||||
TestFile("openai_server/basic/test_openai_server.py", 184),
|
||||
TestFile("openai_server/basic/test_protocol.py", 3),
|
||||
@@ -237,7 +237,7 @@ suite_amd = {
|
||||
TestFile("models/test_qwen_models.py", 82),
|
||||
TestFile("models/test_reward_models.py", 132),
|
||||
TestFile("models/test_transformers_models.py", 320),
|
||||
TestFile("models/test_vlm_models.py", 437),
|
||||
TestFile("models/test_vlm_models.py", 387),
|
||||
TestFile("openai_server/basic/test_openai_embedding.py", 141),
|
||||
TestFile("openai_server/basic/test_openai_server.py", 149),
|
||||
TestFile("openai_server/basic/test_protocol.py", 10),
|
||||
|
||||
Reference in New Issue
Block a user