LanguageBind commited on
Commit
bfcee45
·
verified ·
1 Parent(s): aa27f76

Update opensora/serve/gradio_web_server.py

Browse files
opensora/serve/gradio_web_server.py CHANGED
@@ -24,7 +24,7 @@ from opensora.models.diffusion.latte.modeling_latte import LatteT2V
24
  from opensora.sample.pipeline_videogen import VideoGenPipeline
25
  from opensora.serve.gradio_utils import block_css, title_markdown, randomize_seed_fn, set_env, examples, DESCRIPTION
26
 
27
- import space
28
 
29
  @spaces.GPU
30
  def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
 
24
  from opensora.sample.pipeline_videogen import VideoGenPipeline
25
  from opensora.serve.gradio_utils import block_css, title_markdown, randomize_seed_fn, set_env, examples, DESCRIPTION
26
 
27
+ import spaces
28
 
29
  @spaces.GPU
30
  def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):