From f48d07cd6c5b4d9f8b7b43221b969e5cff73a77f Mon Sep 17 00:00:00 2001
From: hoshi-hiyouga <hiyouga@buaa.edu.cn>
Date: Mon, 25 Nov 2024 19:43:42 +0800
Subject: [PATCH] fix #6136

Former-commit-id: 0516e556a71a22b8767b17734adb94eb127e7e6f
---
 src/llamafactory/model/model_utils/visual.py | 1 +
 1 file changed, 1 insertion(+)

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",