Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "5495073faf05790e48c4e27fe2826b467b69cf4a"
Unverified Commit 100e094c authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Fix autoencoder test (#886) 

Fix autoencoder test.
parent cca59ce3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_models_vae.py
View file @ 100e094c
...@@ -96,7 +96,7 @@ class AutoencoderKLTests(ModelTesterMixin, unittest.TestCase): ...@@ -96,7 +96,7 @@ class AutoencoderKLTests(ModelTesterMixin, unittest.TestCase):
model.config.in_channels, model.config.in_channels,
model.config.sample_size, model.config.sample_size,
model.config.sample_size, model.config.sample_size,
generator=generator, generator=torch.manual_seed(0),
) )
image = image.to(torch_device) image = image.to(torch_device)
with torch.no_grad(): with torch.no_grad():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment