Update app.py

app.py

@@ -188,12 +188,19 @@ class WanInferencePipeline(nn.Module):
         model_manager.load_models(
             [
                 self.args.dit_path.split(","),
-                self.args.vae_path,
                 self.args.text_encoder_path
             ],
             torch_dtype=self.dtype,
             device='cuda',
         )
+
+        model_manager.load_models(
+            [
+                self.args.vae_path,
+            ],
+            torch_dtype=torch.float32,
+            device='cuda',
+        )
         
         pipe = WanVideoPipeline.from_model_manager(model_manager, 
                                                 torch_dtype=self.dtype,