Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -401,10 +401,8 @@ def run_batch(
|
|
| 401 |
top_p: float,
|
| 402 |
max_tokens: int,
|
| 403 |
max_side: int,
|
| 404 |
-
time_budget_s: float
|
| 405 |
-
progress
|
| 406 |
-
if progress is None:
|
| 407 |
-
progress = gr.Progress(track_tqdm=True), # drives the progress bar
|
| 408 |
) -> Tuple[List[dict], list, list, str, List[str], int, int]:
|
| 409 |
"""
|
| 410 |
Returns:
|
|
|
|
| 401 |
top_p: float,
|
| 402 |
max_tokens: int,
|
| 403 |
max_side: int,
|
| 404 |
+
time_budget_s: float = None, # respects Zero-GPU window
|
| 405 |
+
progress = gr.Progress(track_tqdm=True), # drives the progress bar
|
|
|
|
|
|
|
| 406 |
) -> Tuple[List[dict], list, list, str, List[str], int, int]:
|
| 407 |
"""
|
| 408 |
Returns:
|