Spaces:
sonalkum
/
Running on Zero

sonalkum commited on
Commit
1b71190
·
verified ·
1 Parent(s): f21d654

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from utils.prompter import Prompter
16
  import datetime
17
  import time,json
18
 
19
- device = "cuda" if torch.cuda.is_available() else "cpu"
20
 
21
  base_model = "Llama-2-7b-chat-hf-qformer/"
22
 
 
16
  import datetime
17
  import time,json
18
 
19
+ device = "cuda"
20
 
21
  base_model = "Llama-2-7b-chat-hf-qformer/"
22