use new model class for chameleon conversion

This commit is contained in:
nopperl 2024-09-29 10:59:30 +02:00
parent f4d2b8846a
commit 1ddd9ab795

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