From 100e094cc9af523d3f71783bceda82b22b091af2 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Mon, 17 Oct 2022 21:47:13 +0200 Subject: [PATCH] Fix autoencoder test (#886) Fix autoencoder test. --- tests/test_models_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models_vae.py b/tests/test_models_vae.py index 8ade4107bd..9fb7e8ea3b 100644 --- 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():