helloperson123 commited on
Commit
3263c28
·
verified ·
1 Parent(s): b39c5a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model = AutoModelForCausalLM.from_pretrained(
17
  MODEL_NAME,
18
  torch_dtype=torch.float32,
19
 
20
- model.to(DEVICE)
21
 
22
  # ----------------------------
23
  # SYSTEM PROMPT
 
17
  MODEL_NAME,
18
  torch_dtype=torch.float32,
19
 
20
+ model.to(DEVICE))
21
 
22
  # ----------------------------
23
  # SYSTEM PROMPT