Skip to content

Commit 586b164

Browse files
authored
Merge branch 'main' into refactor/condense-group-offloading
2 parents d4f5ec0 + 1082c46 commit 586b164

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)