diff --git a/src/prime_rl/utils/vlm.py b/src/prime_rl/utils/vlm.py index 15b5f1bf3b..a099c19925 100644 --- a/src/prime_rl/utils/vlm.py +++ b/src/prime_rl/utils/vlm.py @@ -28,6 +28,7 @@ class VLMModelInfo: "qwen3_vl": VLMModelInfo(vision_encoder_attr="model.visual", language_model_attr="model.language_model"), "qwen3_5": VLMModelInfo(vision_encoder_attr="model.visual", language_model_attr="model.language_model"), "qwen3_5_moe": VLMModelInfo(vision_encoder_attr="model.visual", language_model_attr="model.language_model"), + "qwen3_vl_moe": VLMModelInfo(vision_encoder_attr="model.visual", language_model_attr="model.language_model"), } # Text-only default