Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
)
|