Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -181,7 +181,7 @@ def process_video_for_control(video, control_type):
|
|
181 |
else:
|
182 |
return video
|
183 |
|
184 |
-
@spaces.GPU(duration=
|
185 |
def generate_video(
|
186 |
reference_video,
|
187 |
prompt,
|
@@ -291,7 +291,7 @@ def generate_video(
|
|
291 |
image_cond_noise_scale=image_cond_noise_scale,
|
292 |
guidance_scale=guidance_scale,
|
293 |
guidance_rescale=guidance_rescale,
|
294 |
-
generator=torch.Generator().manual_seed(seed),
|
295 |
output_type="pil",
|
296 |
).frames[0]
|
297 |
|
|
|
181 |
else:
|
182 |
return video
|
183 |
|
184 |
+
@spaces.GPU(duration=160)
|
185 |
def generate_video(
|
186 |
reference_video,
|
187 |
prompt,
|
|
|
291 |
image_cond_noise_scale=image_cond_noise_scale,
|
292 |
guidance_scale=guidance_scale,
|
293 |
guidance_rescale=guidance_rescale,
|
294 |
+
generator=torch.Generator(device="cuda").manual_seed(seed),
|
295 |
output_type="pil",
|
296 |
).frames[0]
|
297 |
|