Skip to content

Commit 3f36254

Browse files
authored
fix gemma vlm gguf regression (#685)
1 parent 21bb06b commit 3f36254

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

auto_round/export/export_to_gguf/convert.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -767,9 +767,9 @@ def remove_prefix(name, key_list):
767767
if "visual." in name and not name.startswith("visual."):
768768
name = ".".join(name.split(".")[1:])
769769

770-
# for gemma3
770+
# # for gemma3
771771
if self.model_arch == gguf.MODEL_ARCH.GEMMA3 or isinstance(self, Gemma3VisionModel):
772-
visual_keys = ["multi_modal_projector", "vision_tower", "multimodal_projector", "vision_model"]
772+
visual_keys = ["multi_modal_projector", "vision_tower", "multimodal_projector"]
773773
name = remove_prefix(name, visual_keys)
774774

775775
# for LlavaForConditionalGeneration

0 commit comments

Comments
 (0)