Fix test for consistency decoder. (huggingface#7746)

update
SingleZombie · Apr 24, 2024 · 9ef43f3 · 9ef43f3
1 parent 88018fc
commit 9ef43f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/autoencoders/test_models_vae.py b/tests/models/autoencoders/test_models_vae.py
@@ -1153,5 +1153,5 @@ def test_vae_tiling(self):
         shapes = [(1, 4, 73, 97), (1, 4, 97, 73), (1, 4, 49, 65), (1, 4, 65, 49)]
         with torch.no_grad():
             for shape in shapes:
-                image = torch.zeros(shape, device=torch_device)
+                image = torch.zeros(shape, device=torch_device, dtype=pipe.vae.dtype)
                 pipe.vae.decode(image)