Upload folder using huggingface_hub

app.py

@@ -458,7 +458,8 @@ def decode_video(pred_latent):
         1, vae.config.z_dim, 1, 1, 1
     ).to(latents.device, latents.dtype)
     latents = latents / latents_std + latents_mean
-    video = vae.decode(latents, return_dict=False)[0]
+    with torch.no_grad():
+        video = vae.decode(latents, return_dict=False)[0]
     video = video_processor.postprocess_video(video, output_type="np")[0]
     return video
 
@@ -468,7 +469,7 @@ def decode_video(pred_latent):
 # ------------------------------------------------------------------ #
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=120, size="xlarge")
 def generate(
     top_img: "np.ndarray",
     wrist_img: "np.ndarray",