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

This commit is contained in:
nopperl
2024-09-29 12:02:06 +00:00
committed by GitHub
parent 589b48d41e
commit f99d3f8367

View File

@ -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