From ebf753fa553f1d114692187af6de4fff46e71f7f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 29 Dec 2022 13:46:46 +0000 Subject: [PATCH] [StableDiffusionInpaint] Correct test --- .../stable_diffusion_2/test_stable_diffusion_inpaint.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_inpaint.py b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_inpaint.py index 88157f22de6b..756442600ef1 100644 --- a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_inpaint.py +++ b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_inpaint.py @@ -228,7 +228,6 @@ def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): model_id, safety_checker=None, scheduler=pndm, - device_map="auto", torch_dtype=torch.float16, ) pipe.to(torch_device) @@ -244,7 +243,7 @@ def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): image=init_image, mask_image=mask_image, generator=generator, - num_inference_steps=5, + num_inference_steps=2, output_type="np", )