fix #6136

Former-commit-id: b84e5d91a070c473ea820c379bf9b5abbca6df2c
2024-11-25 19:43:42 +08:00
parent b82a53cad8
commit 487e3f2507
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",