Fix autoencoder test (huggingface#886)

pcuenca · Prathik Rao · commit 95d6abb70b26 · 2022-10-26T15:10:08.000-07:00
Fix autoencoder test.
diff --git a/tests/test_models_vae.py b/tests/test_models_vae.py
@@ -96,7 +96,7 @@ def test_output_pretrained(self):
             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():

Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,7 @@ def test_output_pretrained(self):`
`96`	`96`	`model.config.in_channels,`
`97`	`97`	`model.config.sample_size,`
`98`	`98`	`model.config.sample_size,`
`99`		`- generator=generator,`
	`99`	`+ generator=torch.manual_seed(0),`
`100`	`100`	`)`
`101`	`101`	`image = image.to(torch_device)`
`102`	`102`	`with torch.no_grad():`