Spaces:
Sleeping
Sleeping
Commit ·
c1d6742
1
Parent(s): 4b9b37d
fixes
Browse files
app.py
CHANGED
|
@@ -48,11 +48,17 @@ def generate(req: Request):
|
|
| 48 |
outputs = model.generate(
|
| 49 |
**inputs,
|
| 50 |
max_new_tokens=req.max_tokens,
|
|
|
|
| 51 |
do_sample=True,
|
| 52 |
-
temperature=
|
| 53 |
-
top_p=0.
|
| 54 |
-
top_k=40,
|
| 55 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
)
|
| 57 |
|
| 58 |
text = tokenizer.decode(
|
|
|
|
| 48 |
outputs = model.generate(
|
| 49 |
**inputs,
|
| 50 |
max_new_tokens=req.max_tokens,
|
| 51 |
+
|
| 52 |
do_sample=True,
|
| 53 |
+
temperature=0.8,
|
| 54 |
+
top_p=0.92,
|
| 55 |
+
top_k=40,
|
| 56 |
+
|
| 57 |
+
repetition_penalty=1.7, # strong anti-loop
|
| 58 |
+
no_repeat_ngram_size=5, # stronger phrase blocking
|
| 59 |
+
|
| 60 |
+
eos_token_id=tokenizer.eos_token_id,
|
| 61 |
+
pad_token_id=tokenizer.eos_token_id
|
| 62 |
)
|
| 63 |
|
| 64 |
text = tokenizer.decode(
|