Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -136,7 +136,7 @@ def process_video_for_control(video, control_type):
|
|
136 |
else:
|
137 |
return video
|
138 |
|
139 |
-
@spaces.GPU(duration=
|
140 |
def generate_video(
|
141 |
reference_video,
|
142 |
prompt,
|
@@ -246,7 +246,7 @@ def generate_video(
|
|
246 |
image_cond_noise_scale=image_cond_noise_scale,
|
247 |
guidance_scale=guidance_scale,
|
248 |
guidance_rescale=guidance_rescale,
|
249 |
-
generator=torch.Generator().manual_seed(seed),
|
250 |
output_type="pil",
|
251 |
).frames[0]
|
252 |
|
|
|
136 |
else:
|
137 |
return video
|
138 |
|
139 |
+
@spaces.GPU(duration=160)
|
140 |
def generate_video(
|
141 |
reference_video,
|
142 |
prompt,
|
|
|
246 |
image_cond_noise_scale=image_cond_noise_scale,
|
247 |
guidance_scale=guidance_scale,
|
248 |
guidance_rescale=guidance_rescale,
|
249 |
+
generator=torch.Generator(device="cuda").manual_seed(seed),
|
250 |
output_type="pil",
|
251 |
).frames[0]
|
252 |
|