Spaces:
Runtime error
Runtime error
Commit
•
34befce
1
Parent(s):
bab971b
Update opensora/serve/gradio_web_server.py
Browse files
opensora/serve/gradio_web_server.py
CHANGED
|
@@ -24,7 +24,9 @@ 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 |
|
|
|
|
| 28 |
@torch.inference_mode()
|
| 29 |
def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
|
| 30 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
|
|
|
| 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(duration=300)
|
| 30 |
@torch.inference_mode()
|
| 31 |
def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
|
| 32 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|