summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGalunid <karolek1231456@gmail.com>2023-10-23 17:47:03 +0200
committerGitHub <noreply@github.com>2023-10-23 17:47:03 +0200
commit6336701c9378c23c85d1c0e464b663ca2bbb8e60 (patch)
tree7d966ef4c792831dc4a926a55b7d061e02a67707
parent96981f37b1e3f450d9e63e571514217bf60f0a7f (diff)
Fix baichuan convert script not detecing model (#3739)
It seems nobody objects.
-rwxr-xr-xconvert-baichuan-hf-to-gguf.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/convert-baichuan-hf-to-gguf.py b/convert-baichuan-hf-to-gguf.py
index 3b64ecb8..5ee99be7 100755
--- a/convert-baichuan-hf-to-gguf.py
+++ b/convert-baichuan-hf-to-gguf.py
@@ -110,7 +110,7 @@ print("gguf: loading model "+dir_model.name)
with open(dir_model / "config.json", "r", encoding="utf-8") as f:
hparams = json.load(f)
print("hello print: ",hparams["architectures"][0])
-if hparams["architectures"][0] != "BaichuanForCausalLM":
+if hparams["architectures"][0] != "BaichuanForCausalLM" and hparams["architectures"][0] != "BaiChuanForCausalLM":
print("Model architecture not supported: " + hparams["architectures"][0])
sys.exit()