Add Molmo2 multimodal model support (#30997)

Signed-off-by: sanghol <sanghol@allenai.org> Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn> Co-authored-by: Isotr0py <mozf@mail2.sysu.edu.cn>
2026-01-13 23:33:09 -08:00
parent 9312a6c03a
commit 7e6f123810
11 changed files with 3215 additions and 1 deletions
--- a/tests/models/test_initialization.py
+++ b/tests/models/test_initialization.py
@@ -140,6 +140,7 @@ def can_initialize(
            else None,
            trust_remote_code=model_info.trust_remote_code,
            max_model_len=model_info.max_model_len,
+            max_num_batched_tokens=model_info.max_num_batched_tokens,
            # these tests seem to produce leftover memory
            gpu_memory_utilization=0.80,
            load_format="dummy",