ministral3 (#14251)
Signed-off-by: Xinyuan Tong <xinyuantong.cs@gmail.com> Co-authored-by: Yueming Yuan <yy28@illinois.edu>
This commit is contained in:
31
test/srt/models/test_ministral3_models.py
Normal file
31
test/srt/models/test_ministral3_models.py
Normal file
@@ -0,0 +1,31 @@
|
||||
import unittest
|
||||
from types import SimpleNamespace
|
||||
|
||||
from sglang.test.gsm8k_mixin import GSM8KMixin
|
||||
from sglang.test.mmmu_vlm_mixin import MMMUVLMMixin
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
MODEL = "mistralai/Ministral-3-3B-Instruct-2512"
|
||||
|
||||
|
||||
class TestMinistral3TextOnly(GSM8KMixin, CustomTestCase):
|
||||
accuracy = 0.6
|
||||
model = MODEL
|
||||
other_args = ["--trust-remote-code"]
|
||||
|
||||
|
||||
class TestMinistral3MMMU(MMMUVLMMixin, CustomTestCase):
|
||||
accuracy = 0.3
|
||||
model = MODEL
|
||||
other_args = ["--trust-remote-code"]
|
||||
mmmu_args = ["--limit=0.1"]
|
||||
"""`--limit=0.1`: 10 percent of each task - this is fine for testing since the nominal result isn't interesting - this run is just to prevent relative regressions."""
|
||||
|
||||
def test_vlm_mmmu_benchmark(self):
|
||||
self._run_vlm_mmmu_test(
|
||||
SimpleNamespace(model=self.model, mmmu_accuracy=self.accuracy), "./logs"
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@@ -214,6 +214,7 @@ suites = {
|
||||
TestFile("test_moe_eval_accuracy_large.py"),
|
||||
TestFile("test_vision_openai_server_common.py"),
|
||||
TestFile("test_profile_v2.py"),
|
||||
TestFile("models/test_ministral3_models.py"),
|
||||
],
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user