Spaces:

NoaiGPT
/

merger

Build error

NoaiGPT commited on Jul 13

Commit

db0cf1b

•

1 Parent(s): 8295404

asd

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ tokenizer = None
 def merge_models():
     print("Loading base model...")
-    base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
     print("Loading fine-tuned model...")
     fine_tuned_model_path = "NoaiGPT/autotrain-14mrs-fc44l"

 def merge_models():
     print("Loading base model...")
+    base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")
     print("Loading fine-tuned model...")
     fine_tuned_model_path = "NoaiGPT/autotrain-14mrs-fc44l"