Skip to content

Commit f99d3f8

Browse files
authored
py : add model class for Chameleon conversion (ggml-org#9683)
1 parent 589b48d commit f99d3f8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4162,7 +4162,8 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
41624162
return super().modify_tensors(data_torch, name, bid)
41634163

41644164

4165-
@Model.register("ChameleonForCausalLM")
4165+
@Model.register("ChameleonForConditionalGeneration")
4166+
@Model.register("ChameleonForCausalLM") # obsolete
41664167
class ChameleonModel(Model):
41674168
model_arch = gguf.MODEL_ARCH.CHAMELEON
41684169

0 commit comments

Comments
 (0)