From 824cb538b1142e0ca9b7df94e5e6ee100e996109 Mon Sep 17 00:00:00 2001 From: Will Berman Date: Mon, 27 Feb 2023 10:01:59 -0800 Subject: [PATCH] attend and excite batch test causing timeouts (#2498) --- .../test_stable_diffusion_attend_and_excite.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py index 1025b52636..ca300dc0ce 100644 --- a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py +++ b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py @@ -135,8 +135,9 @@ class StableDiffusionAttendAndExcitePipelineFastTests(PipelineTesterMixin, unitt max_diff = np.abs(image_slice.flatten() - expected_slice).max() self.assertLessEqual(max_diff, 1e-3) - def test_inference_batch_single_identical(self): - self._test_inference_batch_single_identical(relax_max_difference=False) + def test_inference_batch_consistent(self): + # NOTE: Larger batch sizes cause this test to timeout, only test on smaller batches + self._test_inference_batch_consistent(batch_sizes=[2, 4]) @require_torch_gpu