Spaces:

mustapha
/

chatAlpaca

Build error

mustapha commited on Mar 18, 2023

Commit

1552f4f

1 Parent(s): 03c78cf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,9 +26,11 @@ model = PeftModel.from_pretrained(
     model, "tloen/alpaca-lora-7b",
     torch_dtype=torch.float16
 )
-model.eval()
 device = "cpu"
 def evaluate_raw_prompt(
         prompt:str,
         temperature=0.1,

     model, "tloen/alpaca-lora-7b",
     torch_dtype=torch.float16
 )
 device = "cpu"
+model.to(device)
+model.eval()
 def evaluate_raw_prompt(
         prompt:str,
         temperature=0.1,