fix #6136

Former-commit-id: 0516e556a71a22b8767b17734adb94eb127e7e6f
2025-08-23 06:12:50 +08:00 · 2024-11-25 19:43:42 +08:00 · 2024-11-25 19:43:42 +08:00 · f48d07cd6c
commit f48d07cd6c
parent f1a1fb675b
1 changed files with 1 additions and 0 deletions
--- a/src/llamafactory/model/model_utils/visual.py
+++ b/src/llamafactory/model/model_utils/visual.py
@ -112,6 +112,7 @@ def configure_visual_model(config: "PretrainedConfig") -> None:
        "llava",
        "llava_next",
        "llava_next_video",
+        "mllama",
        "paligemma",
        "pixtral",
        "video_llava",