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

dsx1986 · Oct 29, 2024 · f4f0ba0 · f4f0ba0
1 parent ef40921
commit f4f0ba0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py
@@ -4162,7 +4162,8 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
         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