Fix autoencoder test (#886)

Fix autoencoder test.

Fix autoencoder test (#886)
Fix autoencoder test.
100e094c · Pedro Cuenca · GitHub · cca59ce3 · 100e094c
Unverified Commit 100e094c authored Oct 17, 2022 by Pedro Cuenca Committed by GitHub Oct 17, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_models_vae.py tests/test_models_vae.py +1 -1

No files found.
--- a/tests/test_models_vae.py
+++ b/tests/test_models_vae.py
@@ -96,7 +96,7 @@ class AutoencoderKLTests(ModelTesterMixin, unittest.TestCase):
            model.config.in_channels,
            model.config.sample_size,
            model.config.sample_size,
-            generator=generator,
+            generator=torch.manual_seed(0),
        )
        image = image.to(torch_device)
        with torch.no_grad():