Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

one more fix

Browse Source
This commit is contained in:
DeepBeepMeep 2025-06-25 09:18:03 +02:00
parent 7676ded155
commit e0666a3e6d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
wan/text2video.py
View File

@ -195,6 +195,7 @@ class WanT2V:
ref_width, ref_height = ref_img.size ref_width, ref_height = ref_img.size
if (ref_height, ref_width) == image_size and outpainting_dims == None: if (ref_height, ref_width) == image_size and outpainting_dims == None:
ref_img = TF.to_tensor(ref_img).sub_(0.5).div_(0.5).unsqueeze(1) ref_img = TF.to_tensor(ref_img).sub_(0.5).div_(0.5).unsqueeze(1)
canvas = torch.zeros_like(ref_img) if return_mask else None
else: else:
if outpainting_dims != None: if outpainting_dims != None:
final_height, final_width = image_size final_height, final_width = image_size