Skip to content

Commit 6336701

Browse files
authored
Fix baichuan convert script not detecing model (#3739)
It seems nobody objects.
1 parent 96981f3 commit 6336701

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert-baichuan-hf-to-gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def parse_args() -> argparse.Namespace:
110110
with open(dir_model / "config.json", "r", encoding="utf-8") as f:
111111
hparams = json.load(f)
112112
print("hello print: ",hparams["architectures"][0])
113-
if hparams["architectures"][0] != "BaichuanForCausalLM":
113+
if hparams["architectures"][0] != "BaichuanForCausalLM" and hparams["architectures"][0] != "BaiChuanForCausalLM":
114114
print("Model architecture not supported: " + hparams["architectures"][0])
115115

116116
sys.exit()

0 commit comments

Comments
 (0)