From 7df73ceaaf0e32b266219f7fcff0dc76ef16b623 Mon Sep 17 00:00:00 2001
From: Zhongjie Duan <35051019+Artiprocher@users.noreply.github.com>
Date: Wed, 26 Feb 2025 20:03:26 +0800
Subject: [PATCH] Fix Wan VAE device

---
 diffsynth/pipelines/wan_video.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/diffsynth/pipelines/wan_video.py b/diffsynth/pipelines/wan_video.py
index f43d559..82ad76c 100644
--- a/diffsynth/pipelines/wan_video.py
+++ b/diffsynth/pipelines/wan_video.py
@@ -95,7 +95,7 @@ class WanVideoPipeline(BasePipeline):
                 offload_dtype=dtype,
                 offload_device="cpu",
                 onload_dtype=dtype,
-                onload_device="cpu",
+                onload_device=self.device,
                 computation_dtype=self.torch_dtype,
                 computation_device=self.device,
             ),
@@ -203,8 +203,8 @@ class WanVideoPipeline(BasePipeline):
         cfg_scale=5.0,
         num_inference_steps=50,
         tiled=True,
-        tile_size=(34, 34),
-        tile_stride=(18, 16),
+        tile_size=(30, 52),
+        tile_stride=(15, 26),
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):