From 2b6c30c54c2a2bdc53c9ed19fc08d8c0e826dd4d Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Mon, 25 Nov 2024 19:43:42 +0800 Subject: [PATCH] fix #6136 Former-commit-id: b84e5d91a070c473ea820c379bf9b5abbca6df2c --- 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",