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, ):