xiangfan00 commited on
Commit
5a9bcbe
·
1 Parent(s): 307d7cf

Increase duration to 160

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -561,7 +561,7 @@ def decode_with_refdecoder(latents, reference_frame, vae, transformer):
561
  return video
562
 
563
 
564
- @spaces.GPU(duration=150)
565
  def generate_latents_on_gpu(image, prompt, seed):
566
  log_cuda_mem("start generate_latents_on_gpu")
567
  GENERATION_PIPE.to(DEVICE)
 
561
  return video
562
 
563
 
564
+ @spaces.GPU(duration=160)
565
  def generate_latents_on_gpu(image, prompt, seed):
566
  log_cuda_mem("start generate_latents_on_gpu")
567
  GENERATION_PIPE.to(DEVICE)