diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_sag.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_sag.py index 96a69d9881..f21a0af5c5 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion_sag.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion_sag.py @@ -136,7 +136,7 @@ class StableDiffusionPipelineIntegrationTests(unittest.TestCase): assert image.shape == (1, 512, 512, 3) expected_slice = np.array([0.1568, 0.1738, 0.1695, 0.1693, 0.1507, 0.1705, 0.1547, 0.1751, 0.1949]) - assert np.abs(image_slice.flatten() - expected_slice).max() < 5e-4 + assert np.abs(image_slice.flatten() - expected_slice).max() < 5e-2 def test_stable_diffusion_2(self): sag_pipe = StableDiffusionSAGPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base") @@ -156,4 +156,4 @@ class StableDiffusionPipelineIntegrationTests(unittest.TestCase): assert image.shape == (1, 512, 512, 3) expected_slice = np.array([0.3459, 0.2876, 0.2537, 0.3002, 0.2671, 0.2160, 0.3026, 0.2262, 0.2371]) - assert np.abs(image_slice.flatten() - expected_slice).max() < 5e-5 + assert np.abs(image_slice.flatten() - expected_slice).max() < 5e-2