prithivMLmods commited on
Commit
b0ec6f1
·
verified ·
1 Parent(s): 6af46dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = 90,
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 90s timeout for examples
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=90
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