Spaces:

adhisetiawan
/

Phi-2-DPO

Runtime error

adhisetiawan commited on Feb 12

Commit

55f8df5

•

1 Parent(s): bd2b1d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from peft import AutoPeftModelForCausalLM
 from threading import Thread
 # Load the fine-tuned model and tokenizer
-new_model = "Ronal999/phi2_DPO"
 model = AutoPeftModelForCausalLM.from_pretrained(new_model,
                                                  low_cpu_mem_usage=True,
                                                  torch_dtype=torch.float16,

 from threading import Thread
 # Load the fine-tuned model and tokenizer
+new_model = "adhisetiawan/phi2_DPO"
 model = AutoPeftModelForCausalLM.from_pretrained(new_model,
                                                  low_cpu_mem_usage=True,
                                                  torch_dtype=torch.float16,