lower cpu usage

app.py

@@ -10,8 +10,7 @@ def generate_video(first_frame_url, last_frame_url, prompt):
     model_id = "Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers"
     image_encoder = CLIPVisionModel.from_pretrained(model_id, subfolder="image_encoder", torch_dtype=torch.float32)
     vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
-    pipe = WanImageToVideoPipeline.from_pretrained(
-        model_id, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16
+    pipe = WanImageToVideoPipeline.from_pretrained(low_cpu_mem_usage=True, model_id, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16
     )
     pipe.to("cuda")