Spaces:

Pr123
/

TinyLlama-EA-Chat

Sleeping

Pr123 commited on Apr 11

Commit

a521714

•

1 Parent(s): 1ea4428

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,6 @@ import gradio as gr
 peft_model_id = "Pr123/TinyLlama-EA-Chat"
-# Load Model with PEFT adapter
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = AutoPeftModelForCausalLM.from_pretrained(
     peft_model_id,

 peft_model_id = "Pr123/TinyLlama-EA-Chat"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = AutoPeftModelForCausalLM.from_pretrained(
     peft_model_id,