Spaces:
Running
Running
GAJJALAASHOKKUMARREDDY commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -539,12 +539,9 @@ Additional Context: {context}
|
|
| 539 |
# Generation parameters
|
| 540 |
generation_kwargs = dict(
|
| 541 |
**inputs,
|
| 542 |
-
max_new_tokens=150, #
|
| 543 |
-
|
| 544 |
-
|
| 545 |
-
top_p=0.95,
|
| 546 |
-
top_k=40,
|
| 547 |
-
repetition_penalty=1.5, # INCREASED to strongly discourage verbosity
|
| 548 |
pad_token_id=phi_tokenizer.eos_token_id,
|
| 549 |
eos_token_id=phi_tokenizer.eos_token_id,
|
| 550 |
streamer=request_streamer,
|
|
|
|
| 539 |
# Generation parameters
|
| 540 |
generation_kwargs = dict(
|
| 541 |
**inputs,
|
| 542 |
+
max_new_tokens=150, # SLIGHTLY LOWER to prevent cutoff
|
| 543 |
+
do_sample=False, # Greedy decoding (no temperature needed)
|
| 544 |
+
repetition_penalty=1.5, # Keep this - prevents repetition
|
|
|
|
|
|
|
|
|
|
| 545 |
pad_token_id=phi_tokenizer.eos_token_id,
|
| 546 |
eos_token_id=phi_tokenizer.eos_token_id,
|
| 547 |
streamer=request_streamer,
|