|
@@ -110,7 +110,7 @@ print("gguf: loading model "+dir_model.name)
|
|
|
with open(dir_model / "config.json", "r", encoding="utf-8") as f:
|
|
with open(dir_model / "config.json", "r", encoding="utf-8") as f:
|
|
|
hparams = json.load(f)
|
|
hparams = json.load(f)
|
|
|
print("hello print: ",hparams["architectures"][0])
|
|
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])
|
|
print("Model architecture not supported: " + hparams["architectures"][0])
|
|
|
|
|
|
|
|
sys.exit()
|
|
sys.exit()
|