Update demo/motiondirector.py

demo/motiondirector.py

@@ -153,8 +153,8 @@ class MotionDirector():
         latents_folder = f"./MotionDirector_pretrained/{model_select_type}/cached_latents"
         latents_path = f"{latents_folder}/{random.choice(os.listdir(latents_folder))}"
         assert os.path.exists(lora_path)
-
         device = "cuda"
+        self.pipe.enable_vae_slicing()
         with torch.autocast(device, dtype=torch.half):
             # prepare input latents
             with torch.no_grad():