Update app.py

app.py

@@ -112,7 +112,7 @@ pathchsageattentionkj = NODE_CLASS_MAPPINGS["PathchSageAttentionKJ"]()
 MODELS_AND_NODES["clip"] = cliploader.load_clip(clip_name="umt5_xxl_fp8_e4m3fn_scaled.safetensors", type="wan")
 unet_low_noise = unetloader.load_unet(unet_name="wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors", weight_dtype="default")
 unet_high_noise = unetloader.load_unet(unet_name="wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors", weight_dtype="default")
-MODELS_AND_NODES["vae"] = vaeloader.load_vae(vae_name="wan_2.1_vae.safetensors", device="cpu")
+MODELS_AND_NODES["vae"] = vaeloader.load_vae(vae_name="wan_2.1_vae.safetensors")
 MODELS_AND_NODES["clip_vision"] = clipvisionloader.load_clip(clip_name="clip_vision_h.safetensors")
 
 # Chain all patching operations together for the final models
@@ -146,7 +146,7 @@ MODELS_AND_NODES["SaveVideo"] = NODE_CLASS_MAPPINGS["SaveVideo"]()
 print("Moving final models to GPU...")
 model_loaders_final = [
     MODELS_AND_NODES["clip"],
-    MODELS_AND_NODES["vae"],
+#    MODELS_AND_NODES["vae"],
     MODELS_AND_NODES["model_low_noise"],
     MODELS_AND_NODES["model_high_noise"],
     MODELS_AND_NODES["clip_vision"],