merge-hf-and-lora-to-hf.py

2023-05-22 19:31:17 +08:00 · 2023-05-22 19:31:17 +08:00 · b78ceb1a2e
commit b78ceb1a2e
parent 29995194e3
1 changed files with 1 additions and 1 deletions
--- a/merge-hf-and-lora-to-hf.py
+++ b/merge-hf-and-lora-to-hf.py
@ -40,7 +40,7 @@ tokenizer.save_pretrained(args.out)

 # load model.
 model = LlamaForCausalLM.from_pretrained(
-    args.model_path,
+    args.model,
    load_in_8bit=False,
    torch_dtype=torch.float16,
    device_map={"": "cpu"},