Skip to content

Commit 359bc7b

Browse files
committed
jpdate test after fixing 5b patchify
1 parent bc7a833 commit 359bc7b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/pipelines/wan/test_wan_22.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -302,8 +302,8 @@ def test_inference(self):
302302
self.assertEqual(generated_video.shape, (9, 3, 32, 32))
303303

304304
# fmt: off
305-
expected_slice = torch.tensor([[0.4814, 0.5336, 0.5094, 0.4922, 0.5061, 0.4923, 0.5043, 0.4923, 0.6821,
306-
0.5965, 0.6753, 0.6014, 0.6939, 0.6076, 0.5133, 0.5651]])
305+
expected_slice = torch.tensor([[[0.4814, 0.4298, 0.5094, 0.4289, 0.5061, 0.4301, 0.5043, 0.4284, 0.5375,
306+
0.5965, 0.5527, 0.6014, 0.5228, 0.6076, 0.6644, 0.5651]]])
307307
# fmt: on
308308

309309
generated_slice = generated_video.flatten()

0 commit comments

Comments
 (0)