ZeroGPU

demo.py

@@ -422,6 +422,7 @@ def generate_animation(
     #     f"Generating animation with EF = {ejection_fraction}, steps = {sampling_steps}, CFG = {cfg_scale}"
     # )
     # print(latent_image.shape, type(latent_image))
+    print("Generating animation...")
 
     if latent_image is None:
         return None
@@ -495,6 +496,8 @@ def generate_animation(
 
     # print("Synthetic video:", synthetic_video.shape)
 
+    print("Animation generated")
+
     return synthetic_video.detach()  # B x C x T x H x W