Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "2aa7712cb72d6f39cc43785a91053cbe0f986843"
Unverified Commit 824cb538 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

attend and excite batch test causing timeouts (#2498)

parent a0549fea
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py
View file @ 824cb538
...@@ -135,8 +135,9 @@ class StableDiffusionAttendAndExcitePipelineFastTests(PipelineTesterMixin, unitt ...@@ -135,8 +135,9 @@ class StableDiffusionAttendAndExcitePipelineFastTests(PipelineTesterMixin, unitt
max_diff = np.abs(image_slice.flatten() - expected_slice).max() max_diff = np.abs(image_slice.flatten() - expected_slice).max()
self.assertLessEqual(max_diff, 1e-3) self.assertLessEqual(max_diff, 1e-3)
def test_inference_batch_single_identical(self): def test_inference_batch_consistent(self):
self._test_inference_batch_single_identical(relax_max_difference=False) # 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 @require_torch_gpu
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment