dar-tau commited on
Commit
dc596e3
·
verified ·
1 Parent(s): 4fb163e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def detach_past_kv(past_kv):
92
  @spaces.GPU
93
  def set_past_key_values():
94
  model, tokenizer = pipe.model, pipe.tokenizer
95
- tokenized = tokenizer(
96
  system_only_prompt_format.format(system_message=system_prompt),
97
  return_tensors='pt'
98
  )
 
92
  @spaces.GPU
93
  def set_past_key_values():
94
  model, tokenizer = pipe.model, pipe.tokenizer
95
+ tokenized = tokenizer.encode(
96
  system_only_prompt_format.format(system_message=system_prompt),
97
  return_tensors='pt'
98
  )