From f3045286f5506b88c86700167f4104a2e03528b4 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Tue, 2 Jan 2024 13:49:56 +0000 Subject: [PATCH] update --- .../stable_diffusion_xl/test_stable_diffusion_xl_img2img.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py index 6d3d42b7a6..97910b124e 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py @@ -784,7 +784,9 @@ class StableDiffusionXLImg2ImgIntegrationTests(unittest.TestCase): "/stable_diffusion_img2img/sketch-mountains-input.png" ) - pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16) + pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained( + "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16 + ) pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config) pipe.unet.set_default_attn_processor() pipe.enable_model_cpu_offload()