From 03bf877bf4e6653137119dd38006f30c76cbee04 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 29 Dec 2022 14:47:56 +0100 Subject: [PATCH] [StableDiffusionInpaint] Correct test (#1859) --- .../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 88157f22de..756442600e 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 @@ class StableDiffusionInpaintPipelineIntegrationTests(unittest.TestCase): model_id, safety_checker=None, scheduler=pndm, - device_map="auto", torch_dtype=torch.float16, ) pipe.to(torch_device) @@ -244,7 +243,7 @@ class StableDiffusionInpaintPipelineIntegrationTests(unittest.TestCase): image=init_image, mask_image=mask_image, generator=generator, - num_inference_steps=5, + num_inference_steps=2, output_type="np", )