Spaces:

helloperson123
/

tiny-llama-chatbot

Runtime error

helloperson123 commited on Jan 11

Commit

3d674e1

verified ·

1 Parent(s): 7cfa400

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,6 @@ CORS(app)
 # Model (UNGATED, FREE)
 # ----------------------------
 model = "microsoft/phi-3-mini-4k-instruct"
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 device = torch.device("cpu")
 model.to(device)

 # Model (UNGATED, FREE)
 # ----------------------------
 model = "microsoft/phi-3-mini-4k-instruct"
 device = torch.device("cpu")
 model.to(device)