Skip to content

Commit 1082c46

Browse files
authored
fix input shape for WanGGUFTexttoVideoSingleFileTests (#12081)
Signed-off-by: jiqing-feng <[email protected]>
1 parent ba2ba90 commit 1082c46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/quantization/gguf/test_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -650,7 +650,7 @@ class WanGGUFTexttoVideoSingleFileTests(GGUFSingleFileTesterMixin, unittest.Test
650650

651651
def get_dummy_inputs(self):
652652
return {
653-
"hidden_states": torch.randn((1, 36, 2, 64, 64), generator=torch.Generator("cpu").manual_seed(0)).to(
653+
"hidden_states": torch.randn((1, 16, 2, 64, 64), generator=torch.Generator("cpu").manual_seed(0)).to(
654654
torch_device, self.torch_dtype
655655
),
656656
"encoder_hidden_states": torch.randn(

0 commit comments

Comments
 (0)