Skip to content

GitLab

Unverified Commit 1082c46a authored by jiqing-feng's avatar jiqing-feng Committed by GitHub
Browse files

fix input shape for WanGGUFTexttoVideoSingleFileTests (#12081) 


Signed-off-by: default avatarjiqing-feng <jiqing.feng@intel.com>
parent ba2ba901
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/quantization/gguf/test_gguf.py
View file @ 1082c46a
...@@ -650,7 +650,7 @@ class WanGGUFTexttoVideoSingleFileTests(GGUFSingleFileTesterMixin, unittest.Test ...@@ -650,7 +650,7 @@ class WanGGUFTexttoVideoSingleFileTests(GGUFSingleFileTesterMixin, unittest.Test
def get_dummy_inputs(self): def get_dummy_inputs(self):
return { return {
"hidden_states": torch.randn((1, 36, 2, 64, 64), generator=torch.Generator("cpu").manual_seed(0)).to( "hidden_states": torch.randn((1, 16, 2, 64, 64), generator=torch.Generator("cpu").manual_seed(0)).to(
torch_device, self.torch_dtype torch_device, self.torch_dtype
), ),
"encoder_hidden_states": torch.randn( "encoder_hidden_states": torch.randn(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment