From: John <redacted>
Date: Fri, 17 Nov 2023 15:24:30 +0000 (+0100)
Subject: py : Falcon HF compatibility (#4104)
X-Git-Tag: upstream/0.0.4488~2957
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=11173c92d6eaa2bd1308c2389f44f838480836ac;p=pkg%2Fggml%2Fsources%2Fllama.cpp

py : Falcon HF compatibility (#4104)

Falcon HF compatibility
---

diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
index e7db7591..3a618fd4 100755
--- 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