Spaces:

adhisetiawan
/

Phi-2-DPO

Runtime error

adhisetiawan commited on Feb 12

Commit

bd2b1d8

•

1 Parent(s): 6c19b81

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,8 @@
 # Install Gradio for creating an interface
 import gradio as gr
 import torch
-from transformers import AutoPeftModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
 from threading import Thread
 # Load the fine-tuned model and tokenizer

 # Install Gradio for creating an interface
 import gradio as gr
 import torch
+from transformers import AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
+from peft import AutoPeftModelForCausalLM
 from threading import Thread
 # Load the fine-tuned model and tokenizer