Skip to content

Commit 332924c

Browse files
convert: register GraniteModel for GraniteSpeechForConditionalGeneration
1 parent ffbc4a9 commit 332924c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

convert_hf_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10590,7 +10590,7 @@ def prepare_tensors(self):
1059010590
raise ValueError(f"Unprocessed experts: {experts}")
1059110591

1059210592

10593-
@ModelBase.register("GraniteForCausalLM")
10593+
@ModelBase.register("GraniteForCausalLM", "GraniteSpeechForConditionalGeneration")
1059410594
class GraniteModel(LlamaModel):
1059510595
"""Conversion for IBM's GraniteForCausalLM"""
1059610596
model_arch = gguf.MODEL_ARCH.GRANITE

0 commit comments

Comments
 (0)