asd
Browse files
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 |
-
|
|
|
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...")
|