Skip to content

Commit 0ba8682

Browse files
committed
fix size
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent d210d4a commit 0ba8682

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/transformers/test_models_transformer_wan_animate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ def get_dummy_inputs(self):
231231
(1, 16, 4, 16, 16), generator=self.generator, device=torch_device, dtype=self.torch_dtype
232232
),
233233
"face_pixel_values": randn_tensor(
234-
(1, 3, 5, 512, 512), generator=self.generator, device=torch_device, dtype=self.torch_dtype
234+
(1, 3, 13, 512, 512), generator=self.generator, device=torch_device, dtype=self.torch_dtype
235235
),
236236
"timestep": torch.tensor([1.0]).to(torch_device, self.torch_dtype),
237237
}

0 commit comments

Comments
 (0)