py : add model class for Chameleon conversion (#9683)
This commit is contained in:
parent
589b48d41e
commit
f99d3f8367
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue