Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -298,7 +298,7 @@ def generate_video(
|
|
| 298 |
randomize_seed: bool = True,
|
| 299 |
height: int = DEFAULT_1_STAGE_HEIGHT,
|
| 300 |
width: int = DEFAULT_1_STAGE_WIDTH,
|
| 301 |
-
gpu_timeout: int =
|
| 302 |
progress=gr.Progress(track_tqdm=True),
|
| 303 |
):
|
| 304 |
gc.collect()
|
|
@@ -408,7 +408,7 @@ def generate_video(
|
|
| 408 |
return None, current_seed
|
| 409 |
|
| 410 |
def generate_video_example(input_image, prompt, lora_adapter, duration):
|
| 411 |
-
# We pass a default
|
| 412 |
output, seed = generate_video(
|
| 413 |
input_image=input_image,
|
| 414 |
prompt=prompt,
|
|
@@ -419,7 +419,7 @@ def generate_video_example(input_image, prompt, lora_adapter, duration):
|
|
| 419 |
randomize_seed=True,
|
| 420 |
height=DEFAULT_1_STAGE_HEIGHT,
|
| 421 |
width=DEFAULT_1_STAGE_WIDTH,
|
| 422 |
-
gpu_timeout=
|
| 423 |
)
|
| 424 |
return output
|
| 425 |
|
|
|
|
| 298 |
randomize_seed: bool = True,
|
| 299 |
height: int = DEFAULT_1_STAGE_HEIGHT,
|
| 300 |
width: int = DEFAULT_1_STAGE_WIDTH,
|
| 301 |
+
gpu_timeout: int = 120,
|
| 302 |
progress=gr.Progress(track_tqdm=True),
|
| 303 |
):
|
| 304 |
gc.collect()
|
|
|
|
| 408 |
return None, current_seed
|
| 409 |
|
| 410 |
def generate_video_example(input_image, prompt, lora_adapter, duration):
|
| 411 |
+
# We pass a default 120s timeout for examples
|
| 412 |
output, seed = generate_video(
|
| 413 |
input_image=input_image,
|
| 414 |
prompt=prompt,
|
|
|
|
| 419 |
randomize_seed=True,
|
| 420 |
height=DEFAULT_1_STAGE_HEIGHT,
|
| 421 |
width=DEFAULT_1_STAGE_WIDTH,
|
| 422 |
+
gpu_timeout=120
|
| 423 |
)
|
| 424 |
return output
|
| 425 |
|