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