Просмотр исходного кода

convert : add qwen2vl support for unsloth merges (#13686)

antichristHater 8 месяцев назад
Родитель
Сommit
c76532e7ba
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      convert_hf_to_gguf.py

+ 2 - 2
convert_hf_to_gguf.py

@@ -2645,7 +2645,7 @@ class Qwen2Model(TextModel):
         yield from super().modify_tensors(data_torch, name, bid)
         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):
 class Qwen2VLModel(TextModel):
     model_arch = gguf.MODEL_ARCH.QWEN2VL
     model_arch = gguf.MODEL_ARCH.QWEN2VL
 
 
@@ -2669,7 +2669,7 @@ class Qwen2VLModel(TextModel):
         return [(self.map_tensor_name(name), data_torch)]
         return [(self.map_tensor_name(name), data_torch)]
 
 
 
 
-@ModelBase.register("Qwen2VLForConditionalGeneration", "Qwen2_5_VLForConditionalGeneration")
+@ModelBase.register("Qwen2VLModel", "Qwen2VLForConditionalGeneration", "Qwen2_5_VLForConditionalGeneration")
 class Qwen2VLVisionModel(VisionModel):
 class Qwen2VLVisionModel(VisionModel):
     def __init__(self, *args, **kwargs):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         super().__init__(*args, **kwargs)