Update app.py

app.py

@@ -14,8 +14,12 @@ MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 LORA_REPO_ID = "Kijai/WanVideo_comfy"
 LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32_v2.safetensors"
 
-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)
+image_encoder = CLIPVisionModel.from_pretrained(
+    MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float16
+)
+vae = AutoencoderKLWan.from_pretrained(
+    MODEL_ID, subfolder="vae", torch_dtype=torch.float16
+)
 pipe = WanImageToVideoPipeline.from_pretrained(
     MODEL_ID, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16
 )