diff options
| author | John <78893154+cmp-nct@users.noreply.github.com> | 2023-11-17 16:24:30 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-17 17:24:30 +0200 |
| commit | 11173c92d6eaa2bd1308c2389f44f838480836ac (patch) | |
| tree | beadd5fd7a0bce417c87e7f5959c34a7d20bf03b | |
| parent | 9e87ef60e18d69338c5efea314aa7e718bf2040a (diff) | |
py : Falcon HF compatibility (#4104)
Falcon HF compatibility
| -rwxr-xr-x | convert-hf-to-gguf.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |