ZeroGPU
Browse files
demo.py
CHANGED
|
@@ -498,6 +498,8 @@ def generate_animation(
|
|
| 498 |
return synthetic_video.detach() # B x C x T x H x W
|
| 499 |
|
| 500 |
|
|
|
|
|
|
|
| 501 |
def decode_animation(latent_animation):
|
| 502 |
"""Decode a latent animation to pixel space"""
|
| 503 |
if latent_animation is None:
|
|
|
|
| 498 |
return synthetic_video.detach() # B x C x T x H x W
|
| 499 |
|
| 500 |
|
| 501 |
+
@spaces.GPU
|
| 502 |
+
@torch.no_grad()
|
| 503 |
def decode_animation(latent_animation):
|
| 504 |
"""Decode a latent animation to pixel space"""
|
| 505 |
if latent_animation is None:
|