Hugging Face's logo

Spaces:
codys12
/
MergeLlama-7b
Paused

App Files Community
2
codys12 commited on
Commit
5b4300b
•
1 Parent(s): 51a11ce

cuda:0

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def generate(
45
  history = current_input
46
  current_input += message
47
 
48
- device = "cuda"
49
  print(current_input)
50
  input_ids = tokenizer(current_input, return_tensors="pt").input_ids.to(device)
51
 
 
45
  history = current_input
46
  current_input += message
47
 
48
+ device = "cuda:0"
49
  print(current_input)
50
  input_ids = tokenizer(current_input, return_tensors="pt").input_ids.to(device)
51