From 2a2d4ad21fb722c37bc640e015320eec45228222 Mon Sep 17 00:00:00 2001 From: Sebastian Gilits Date: Wed, 4 Oct 2023 16:13:23 +0200 Subject: [PATCH 1/3] Reduce number of down block channels --- .../stable_diffusion_xl/test_stable_diffusion_xl.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py index 65c7526e3aa2..00dff1a3d4ba 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py @@ -51,7 +51,7 @@ class StableDiffusionXLPipelineFastTests(PipelineLatentTesterMixin, PipelineTest def get_dummy_components(self): torch.manual_seed(0) unet = UNet2DConditionModel( - block_out_channels=(32, 64), + block_out_channels=(2, 4), layers_per_block=2, sample_size=32, in_channels=4, @@ -66,6 +66,7 @@ def get_dummy_components(self): transformer_layers_per_block=(1, 2), projection_class_embeddings_input_dim=80, # 6 * 8 + 32 cross_attention_dim=64, + norm_num_groups=1, ) scheduler = EulerDiscreteScheduler( beta_start=0.00085, @@ -145,8 +146,10 @@ def test_stable_diffusion_xl_euler(self): assert image.shape == (1, 64, 64, 3) expected_slice = np.array([0.5873, 0.6128, 0.4797, 0.5122, 0.5674, 0.4639, 0.5227, 0.5149, 0.4747]) + max_deviation = np.abs(image_slice.flatten() - expected_slice).max() + print(max_deviation) - assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2 + assert max_deviation < 1e-1 def test_stable_diffusion_xl_prompt_embeds(self): components = self.get_dummy_components() From cf040187b607e89028f95b5298d0b5e1dcfe4418 Mon Sep 17 00:00:00 2001 From: Sebastian Gilits Date: Wed, 4 Oct 2023 16:15:00 +0200 Subject: [PATCH 2/3] Remove debug code --- .../pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py index 00dff1a3d4ba..b05f36668ba0 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py @@ -146,10 +146,8 @@ def test_stable_diffusion_xl_euler(self): assert image.shape == (1, 64, 64, 3) expected_slice = np.array([0.5873, 0.6128, 0.4797, 0.5122, 0.5674, 0.4639, 0.5227, 0.5149, 0.4747]) - max_deviation = np.abs(image_slice.flatten() - expected_slice).max() - print(max_deviation) - assert max_deviation < 1e-1 + assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-1 def test_stable_diffusion_xl_prompt_embeds(self): components = self.get_dummy_components() From b658ed01c27011c16097d0e2894c33675f31dc20 Mon Sep 17 00:00:00 2001 From: Sebastian Gilits Date: Sun, 8 Oct 2023 11:23:01 +0200 Subject: [PATCH 3/3] Set new excepted image slice values for sdxl euler test --- .../pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py index b05f36668ba0..cebd860a4379 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py @@ -145,9 +145,9 @@ def test_stable_diffusion_xl_euler(self): image_slice = image[0, -3:, -3:, -1] assert image.shape == (1, 64, 64, 3) - expected_slice = np.array([0.5873, 0.6128, 0.4797, 0.5122, 0.5674, 0.4639, 0.5227, 0.5149, 0.4747]) + expected_slice = np.array([0.5552, 0.5569, 0.4725, 0.4348, 0.4994, 0.4632, 0.5142, 0.5012, 0.47]) - assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-1 + assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2 def test_stable_diffusion_xl_prompt_embeds(self): components = self.get_dummy_components()