ZeroGPU
Browse files
demo.py
CHANGED
|
@@ -573,6 +573,8 @@ def convert_latent_to_display(latent_image):
|
|
| 573 |
return display_image
|
| 574 |
|
| 575 |
|
|
|
|
|
|
|
| 576 |
def latent_animation_to_grayscale(latent_animation):
|
| 577 |
"""Convert multi-channel latent animation to grayscale for display"""
|
| 578 |
if latent_animation is None:
|
|
|
|
| 573 |
return display_image
|
| 574 |
|
| 575 |
|
| 576 |
+
@spaces.GPU
|
| 577 |
+
@torch.no_grad()
|
| 578 |
def latent_animation_to_grayscale(latent_animation):
|
| 579 |
"""Convert multi-channel latent animation to grayscale for display"""
|
| 580 |
if latent_animation is None:
|