Hugging Face's logo

Spaces:
codys12
/
MergeLlama-7b
Paused

App Files Community
2
codys12 commited on
Commit
1ba36bf
•
1 Parent(s): 9a7bc17

AAAH

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