Skip to content

Commit 11173c9

Browse files
authored
py : Falcon HF compatibility (#4104)
Falcon HF compatibility
1 parent 9e87ef6 commit 11173c9

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
@@ -193,7 +193,7 @@ def _get_model_architecture(self) -> gguf.MODEL_ARCH:
193193
return gguf.MODEL_ARCH.MPT
194194
if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
195195
return gguf.MODEL_ARCH.BAICHUAN
196-
if arch == "FalconForCausalLM":
196+
if arch in ("FalconForCausalLM", "RWForCausalLM"):
197197
return gguf.MODEL_ARCH.FALCON
198198
if arch == "GPTBigCodeForCausalLM":
199199
return gguf.MODEL_ARCH.STARCODER

0 commit comments

Comments
 (0)