Skip to content

Commit 047032d

Browse files
committed
Duh - add llava in another place
1 parent 9e035cd commit 047032d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert-hf-to-gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ def _get_model_architecture(self) -> gguf.MODEL_ARCH:
201201
return gguf.MODEL_ARCH.REFACT
202202
if arch == "PersimmonForCausalLM":
203203
return gguf.MODEL_ARCH.PERSIMMON
204-
if arch == "StableLMEpochForCausalLM":
204+
if arch in ("StableLMEpochForCausalLM", "LlavaStableLMEpochForCausalLM"):
205205
return gguf.MODEL_ARCH.STABLELM
206206

207207
raise NotImplementedError(f'Architecture "{arch}" not supported!')

0 commit comments

Comments
 (0)