From e0666a3e6d8a89c82d97275fd445a0e1f3b4486a Mon Sep 17 00:00:00 2001
From: DeepBeepMeep <deepbeepmeep@yahoo.com>
Date: Wed, 25 Jun 2025 09:18:03 +0200
Subject: [PATCH] one more fix

---
 wan/text2video.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/wan/text2video.py b/wan/text2video.py
index 03ceef4..dfa456b 100644
--- a/wan/text2video.py
+++ b/wan/text2video.py
@@ -195,6 +195,7 @@ class WanT2V:
         ref_width, ref_height = ref_img.size
         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)
+            canvas = torch.zeros_like(ref_img) if return_mask else None
         else:
             if outpainting_dims != None:
                 final_height, final_width = image_size