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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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