diff --git a/src/llamafactory/model/model_utils/visual.py b/src/llamafactory/model/model_utils/visual.py index 04f1eae8..4204c6a4 100644 --- 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",