LanguageBind commited on
Commit
2ba42e0
1 Parent(s): f803893

Update opensora/serve/gradio_web_server.py

Browse files
opensora/serve/gradio_web_server.py CHANGED
@@ -26,7 +26,7 @@ from opensora.serve.gradio_utils import block_css, title_markdown, randomize_see
26
 
27
  import spaces
28
 
29
- @spaces.GPU
30
  def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
31
  seed = int(randomize_seed_fn(seed, randomize_seed))
32
  set_env(seed)
 
26
 
27
  import spaces
28
 
29
+ @spaces.GPU(duration=250)
30
  def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
31
  seed = int(randomize_seed_fn(seed, randomize_seed))
32
  set_env(seed)