Adjust default video generation parameters and GPU decorator
Browse files
app.py
CHANGED
@@ -65,7 +65,7 @@ def save_video(video_frames, fps=25):
|
|
65 |
export_to_video(video_frames, tmp_file.name, fps=fps)
|
66 |
return tmp_file.name
|
67 |
|
68 |
-
@spaces.GPU(
|
69 |
def generate_video_pipeline(pipeline, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
|
70 |
"""Generate video using the pipeline"""
|
71 |
# Move to appropriate device
|
@@ -140,7 +140,7 @@ def create_interface():
|
|
140 |
num_frames = gr.Slider(
|
141 |
minimum=8,
|
142 |
maximum=128,
|
143 |
-
value=
|
144 |
step=1,
|
145 |
label="Number of Frames",
|
146 |
info="Number of frames in the video"
|
@@ -164,7 +164,7 @@ def create_interface():
|
|
164 |
num_inference_steps = gr.Slider(
|
165 |
minimum=10,
|
166 |
maximum=100,
|
167 |
-
value=
|
168 |
step=1,
|
169 |
label="Inference Steps",
|
170 |
info="Higher number = better quality but slower"
|
|
|
65 |
export_to_video(video_frames, tmp_file.name, fps=fps)
|
66 |
return tmp_file.name
|
67 |
|
68 |
+
@spaces.GPU()
|
69 |
def generate_video_pipeline(pipeline, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
|
70 |
"""Generate video using the pipeline"""
|
71 |
# Move to appropriate device
|
|
|
140 |
num_frames = gr.Slider(
|
141 |
minimum=8,
|
142 |
maximum=128,
|
143 |
+
value=40,
|
144 |
step=1,
|
145 |
label="Number of Frames",
|
146 |
info="Number of frames in the video"
|
|
|
164 |
num_inference_steps = gr.Slider(
|
165 |
minimum=10,
|
166 |
maximum=100,
|
167 |
+
value=40,
|
168 |
step=1,
|
169 |
label="Inference Steps",
|
170 |
info="Higher number = better quality but slower"
|