Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,8 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
|
|
49 |
max_new_tokens=max_new_tokens,
|
50 |
temperature=temperature,
|
51 |
return_full_text = False,
|
52 |
-
forced_eos_token_id=tokenizer.encode("[...]")[1]
|
|
|
53 |
)
|
54 |
generated_text = output[0]['generated_text']
|
55 |
|
|
|
49 |
max_new_tokens=max_new_tokens,
|
50 |
temperature=temperature,
|
51 |
return_full_text = False,
|
52 |
+
forced_eos_token_id=tokenizer.encode("[...]")[1],
|
53 |
+
pad_token_id=tokenizer.eos_token_id
|
54 |
)
|
55 |
generated_text = output[0]['generated_text']
|
56 |
|