Spaces:
Sleeping
Sleeping
use old token function
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def create_response_original(input_str,
|
|
39 |
# num_return_sequences=num_return_sequences)[0])
|
40 |
|
41 |
input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
|
42 |
-
output_ids = original_model.generate(
|
43 |
do_sample=do_sample,
|
44 |
max_length=100,
|
45 |
temperature=temperature,
|
|
|
39 |
# num_return_sequences=num_return_sequences)[0])
|
40 |
|
41 |
input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
|
42 |
+
output_ids = original_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200),
|
43 |
do_sample=do_sample,
|
44 |
max_length=100,
|
45 |
temperature=temperature,
|