Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
|
@@ -429,7 +429,7 @@ def get_beam_search_html(
|
|
| 429 |
**inputs,
|
| 430 |
max_new_tokens=number_steps,
|
| 431 |
num_beams=number_beams,
|
| 432 |
-
num_return_sequences=num_return_sequences,
|
| 433 |
return_dict_in_generate=True,
|
| 434 |
length_penalty=length_penalty,
|
| 435 |
output_scores=True,
|
|
|
|
| 429 |
**inputs,
|
| 430 |
max_new_tokens=number_steps,
|
| 431 |
num_beams=number_beams,
|
| 432 |
+
num_return_sequences=min(num_return_sequences, number_beams),
|
| 433 |
return_dict_in_generate=True,
|
| 434 |
length_penalty=length_penalty,
|
| 435 |
output_scores=True,
|