Spaces:
Runtime error
Runtime error
Commit
•
c402338
1
Parent(s):
9e5f079
Update app.py
Browse files
app.py
CHANGED
@@ -32,6 +32,8 @@ def inference_func(prompt, max_length=64, sampling_temperature=0.7):
|
|
32 |
max_length=max_length,
|
33 |
sampling_topk=20,
|
34 |
sampling_temperature=sampling_temperature,
|
|
|
|
|
35 |
include_prompt_in_result=False,
|
36 |
)
|
37 |
output = tokenizer.decode(results[0].sequences_ids[0])
|
|
|
32 |
max_length=max_length,
|
33 |
sampling_topk=20,
|
34 |
sampling_temperature=sampling_temperature,
|
35 |
+
repetition_penalty=1.1,
|
36 |
+
end_token=[26168, 27, 208, 14719, 9078, 18482, 27, 208],
|
37 |
include_prompt_in_result=False,
|
38 |
)
|
39 |
output = tokenizer.decode(results[0].sequences_ids[0])
|