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 |
|