py : Falcon HF compatibility (#4104)

Falcon HF compatibility

diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
index e7db7591260af82ddd5d826181147ff4a1c4f7b3..3a618fd4dc0f65e75b61e9f8c93b082732f21664 100755 (executable)
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@@ -193,7 +193,7 @@ class Model:
             return gguf.MODEL_ARCH.MPT
         if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
             return gguf.MODEL_ARCH.BAICHUAN
-        if arch == "FalconForCausalLM":
+        if arch in ("FalconForCausalLM", "RWForCausalLM"):
             return gguf.MODEL_ARCH.FALCON
         if arch == "GPTBigCodeForCausalLM":
             return gguf.MODEL_ARCH.STARCODER