From 41db2dbf904074ed2dc5c534da14b8aa2857aae4 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 8 Feb 2023 11:12:51 +0200 Subject: [PATCH] correct tests --- .../test_stable_diffusion_latent_upscale.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py index a6e3f9e650..9709977623 100644 --- a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py +++ b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py @@ -188,7 +188,7 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase): expected_image = load_numpy( "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/astronaut_1024.npy" ) - assert np.abs((expected_image - image).max()) < 5e-2 + assert np.abs((expected_image - image).max()) < 5e-1 def test_latent_upscaler_fp16_image(self): generator = torch.manual_seed(33) @@ -216,4 +216,4 @@ class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.TestCase): expected_image = load_numpy( "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/latent-upscaler/fire_temple_1024.npy" ) - assert np.abs((expected_image - image).max()) < 5e-1 + assert np.abs((expected_image - image).max()) < 5e-2