From 2dc25c150ec4288e0b8689fc49fe2c9c8ab01999 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 11 Apr 2025 00:56:54 +0300 Subject: [PATCH] allow other dimensions --- wanvideo/modules/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index 8b010e4..dc18e77 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -473,7 +473,7 @@ class WanAttentionBlock(nn.Module): # encode ReCamMaster camera camera_embed = self.cam_encoder(camera_embed.to(x)) camera_embed = camera_embed.repeat(1, 2, 1) - camera_embed = camera_embed.unsqueeze(2).unsqueeze(3).repeat(1, 1, 30, 52, 1) + camera_embed = camera_embed.unsqueeze(2).unsqueeze(3).repeat(1, 1, grid_sizes[0][1], grid_sizes[0][2], 1) camera_embed = rearrange(camera_embed, 'b f h w d -> b (f h w) d') input_x += camera_embed