NoaiGPT commited on
Commit
f950f71
1 Parent(s): 0bc9204
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ def merge_models():
14
  base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
15
 
16
  print("Loading fine-tuned model...")
17
- peft_config = PeftConfig.from_pretrained("NoaiGPT/autotrain-14mrs-fc44l")
 
18
  fine_tuned_model = PeftModel.from_pretrained(base_model, peft_config)
19
 
20
  print("Merging models...")
 
14
  base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
15
 
16
  print("Loading fine-tuned model...")
17
+ fine_tuned_model_path = "NoaiGPT/autotrain-14mrs-fc44l"
18
+ peft_config = PeftConfig.from_pretrained(fine_tuned_model_path)
19
  fine_tuned_model = PeftModel.from_pretrained(base_model, peft_config)
20
 
21
  print("Merging models...")