theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-23 09:28:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Wan VAE device

Browse Source
This commit is contained in:
Zhongjie Duan
2025-02-26 20:03:26 +08:00
committed by GitHub
parent cf12723c89
commit 7df73ceaaf
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
diffsynth/pipelines/wan_video.py
View File

@@ -95,7 +95,7 @@ class WanVideoPipeline(BasePipeline):
offload_dtype=dtype, offload_dtype=dtype,
offload_device="cpu", offload_device="cpu",
onload_dtype=dtype, onload_dtype=dtype,
onload_device="cpu", onload_device=self.device,
computation_dtype=self.torch_dtype, computation_dtype=self.torch_dtype,
computation_device=self.device, computation_device=self.device,
), ),
@@ -203,8 +203,8 @@ class WanVideoPipeline(BasePipeline):
cfg_scale=5.0, cfg_scale=5.0,
num_inference_steps=50, num_inference_steps=50,
tiled=True, tiled=True,
tile_size=(34, 34), tile_size=(30, 52),
tile_stride=(18, 16), tile_stride=(15, 26),
progress_bar_cmd=tqdm, progress_bar_cmd=tqdm,
progress_bar_st=None, progress_bar_st=None,
): ):