Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import snapshot_download
|
|
7 |
from pyramid_dit import PyramidDiTForVideoGeneration
|
8 |
from diffusers.utils import export_to_video
|
9 |
|
10 |
-
import spaces
|
11 |
import uuid
|
12 |
|
13 |
is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow" else False
|
@@ -61,7 +61,7 @@ def load_model():
|
|
61 |
model = load_model()
|
62 |
|
63 |
# Text-to-video generation function
|
64 |
-
@spaces.GPU(duration=140)
|
65 |
def generate_video(prompt, image=None, duration=3, guidance_scale=9, video_guidance_scale=5, frames_per_second=8, progress=gr.Progress(track_tqdm=True)):
|
66 |
multiplier = 1.2 if is_canonical else 3.0
|
67 |
temp = int(duration * multiplier) + 1
|
|
|
7 |
from pyramid_dit import PyramidDiTForVideoGeneration
|
8 |
from diffusers.utils import export_to_video
|
9 |
|
10 |
+
# import spaces
|
11 |
import uuid
|
12 |
|
13 |
is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow" else False
|
|
|
61 |
model = load_model()
|
62 |
|
63 |
# Text-to-video generation function
|
64 |
+
# @spaces.GPU(duration=140)
|
65 |
def generate_video(prompt, image=None, duration=3, guidance_scale=9, video_guidance_scale=5, frames_per_second=8, progress=gr.Progress(track_tqdm=True)):
|
66 |
multiplier = 1.2 if is_canonical else 3.0
|
67 |
temp = int(duration * multiplier) + 1
|