Fix baichuan convert script not detecing model (ggerganov#3739)

It seems nobody objects.
hodlen · Oct 23, 2023 · 6336701 · 6336701
1 parent 96981f3
commit 6336701
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/convert-baichuan-hf-to-gguf.py b/convert-baichuan-hf-to-gguf.py
@@ -110,7 +110,7 @@ def parse_args() -> argparse.Namespace:
 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()