theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-18 22:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #352 from modelscope/bugfix

Browse Source 
Fix Wan VAE device
This commit is contained in:
Zhongjie Duan
2025-02-26 20:03:53 +08:00
committed by GitHub
parent cf12723c89 7df73ceaaf
commit 1d309a14a3
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_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,
):