Fix autoencoder test (#886)

Fix autoencoder test.
2026-01-27 17:22:53 +03:00 · 2022-10-17 21:47:13 +02:00
parent cca59ce3a2
commit 100e094cc9
1 changed files with 1 additions and 1 deletions
--- 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():