Browse Source

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

nopperl 1 year ago
parent
commit
f99d3f8367
1 changed files with 2 additions and 1 deletions
  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)
         return super().modify_tensors(data_torch, name, bid)
 
 
 
 
-@Model.register("ChameleonForCausalLM")
+@Model.register("ChameleonForConditionalGeneration")
+@Model.register("ChameleonForCausalLM")  # obsolete
 class ChameleonModel(Model):
 class ChameleonModel(Model):
     model_arch = gguf.MODEL_ARCH.CHAMELEON
     model_arch = gguf.MODEL_ARCH.CHAMELEON