From 0a0e2cd1821150550a23cef52642bb54b7fb94c0 Mon Sep 17 00:00:00 2001 From: charchit7 Date: Wed, 4 Oct 2023 19:32:06 +0530 Subject: [PATCH] TODO:changes to architecture internals for fast test --- tests/pipelines/stable_diffusion/test_stable_diffusion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion.py b/tests/pipelines/stable_diffusion/test_stable_diffusion.py index df9e8d47f1b1..6ac7cc0c531b 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion.py @@ -90,6 +90,8 @@ def _test_stable_diffusion_compile(in_queue, out_queue, timeout): out_queue.put(results, timeout=timeout) out_queue.join() +# the issue is to make fast test for StableDiffusionPipeline smaller! +# TODO : changes to the architecture of UNET/CLIP/VAE class StableDiffusionPipelineFastTests( PipelineLatentTesterMixin, PipelineKarrasSchedulerTesterMixin, PipelineTesterMixin, unittest.TestCase