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

py : add model class for Chameleon conversion (#9683)

nopperl 1 год назад
Родитель
Сommit
f99d3f8367
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      convert_hf_to_gguf.py

+ 2 - 1
convert_hf_to_gguf.py

@@ -4162,7 +4162,8 @@ class GraniteMoeModel(GraniteModel):
         return super().modify_tensors(data_torch, name, bid)
 
 
-@Model.register("ChameleonForCausalLM")
+@Model.register("ChameleonForConditionalGeneration")
+@Model.register("ChameleonForCausalLM")  # obsolete
 class ChameleonModel(Model):
     model_arch = gguf.MODEL_ARCH.CHAMELEON