Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-07 15:54:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix MPS compatibility for I2V by adjusting device usage and dtype

Browse Source
This commit is contained in:
Bakhtiyor Sulaymonov 2025-02-27 18:55:10 +05:00
parent 68ae7184df
commit 5cb67c67e8
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wan/configs/wan_i2v_14B.py
View File

@ -14,7 +14,7 @@ i2v_14B.t5_tokenizer = 'google/umt5-xxl'
# clip # clip
i2v_14B.clip_model = 'clip_xlm_roberta_vit_h_14' i2v_14B.clip_model = 'clip_xlm_roberta_vit_h_14'
i2v_14B.clip_dtype = torch.float16 i2v_14B.clip_dtype = torch.float32
i2v_14B.clip_checkpoint = 'models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth' i2v_14B.clip_checkpoint = 'models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth'
i2v_14B.clip_tokenizer = 'xlm-roberta-large' i2v_14B.clip_tokenizer = 'xlm-roberta-large'

2
wan/image2video.py
View File

@ -244,7 +244,7 @@ class WanI2V:
y = self.vae.encode([ y = self.vae.encode([
torch.concat([ torch.concat([
torch.nn.functional.interpolate( torch.nn.functional.interpolate(
img[None].cpu(), size=(h, w), mode='bicubic').transpose( img[None], size=(h, w), mode='bicubic').transpose(
0, 1), 0, 1),
torch.zeros(3, 80, h, w) torch.zeros(3, 80, h, w)
], ],