From c76532e7ba128bb097bf6836bf0f5592e1b56b76 Mon Sep 17 00:00:00 2001 From: antichristHater <142441588+antichristHater@users.noreply.github.com> Date: Wed, 21 May 2025 19:40:35 +0300 Subject: [PATCH] convert : add qwen2vl support for unsloth merges (#13686) --- convert_hf_to_gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 15e019a1..e88076cc 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -2645,7 +2645,7 @@ class Qwen2Model(TextModel): yield from super().modify_tensors(data_torch, name, bid) -@ModelBase.register("Qwen2VLForConditionalGeneration", "Qwen2_5_VLForConditionalGeneration") +@ModelBase.register("Qwen2VLModel", "Qwen2VLForConditionalGeneration", "Qwen2_5_VLForConditionalGeneration") class Qwen2VLModel(TextModel): model_arch = gguf.MODEL_ARCH.QWEN2VL @@ -2669,7 +2669,7 @@ class Qwen2VLModel(TextModel): return [(self.map_tensor_name(name), data_torch)] -@ModelBase.register("Qwen2VLForConditionalGeneration", "Qwen2_5_VLForConditionalGeneration") +@ModelBase.register("Qwen2VLModel", "Qwen2VLForConditionalGeneration", "Qwen2_5_VLForConditionalGeneration") class Qwen2VLVisionModel(VisionModel): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs)