Skip to content

Improve performance of fast test by reducing down blocks #5290

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Oct 9, 2023
Merged

Improve performance of fast test by reducing down blocks #5290

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
2a2d4ad
Reduce number of down block channels
sepal Oct 4, 2023
cf04018
Remove debug code
sepal Oct 4, 2023
0d8e57e
Merge branch 'main' into faster-fast-tests-reduce-down-blocks
sepal Oct 5, 2023
e619e36
Merge branch 'main' into faster-fast-tests-reduce-down-blocks
sepal Oct 5, 2023
0eba8aa
Merge branch 'main' into faster-fast-tests-reduce-down-blocks
sepal Oct 6, 2023
b658ed0
Set new excepted image slice values for sdxl euler test
sepal Oct 8, 2023
d4d210a
Merge branch 'main' into faster-fast-tests-reduce-down-blocks
sepal Oct 8, 2023
cad2c74
Merge branch 'main' into faster-fast-tests-reduce-down-blocks
sepal Oct 9, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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,
Expand Down Expand Up @@ -144,7 +145,7 @@ 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-2

Expand Down