瀏覽代碼

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

nopperl 1 年之前
父節點
當前提交
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