Browse Source

py : Falcon HF compatibility (#4104)

Falcon HF compatibility
John 2 years ago
parent
commit
11173c92d6
1 changed files with 1 additions and 1 deletions
  1. 1 1
      convert-hf-to-gguf.py

+ 1 - 1
convert-hf-to-gguf.py

@@ -193,7 +193,7 @@ class Model:
             return gguf.MODEL_ARCH.MPT
             return gguf.MODEL_ARCH.MPT
         if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
         if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
             return gguf.MODEL_ARCH.BAICHUAN
             return gguf.MODEL_ARCH.BAICHUAN
-        if arch == "FalconForCausalLM":
+        if arch in ("FalconForCausalLM", "RWForCausalLM"):
             return gguf.MODEL_ARCH.FALCON
             return gguf.MODEL_ARCH.FALCON
         if arch == "GPTBigCodeForCausalLM":
         if arch == "GPTBigCodeForCausalLM":
             return gguf.MODEL_ARCH.STARCODER
             return gguf.MODEL_ARCH.STARCODER