From 8dd86c9bd9798faa47919bc6531ed7ba483e2f32 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radam=C3=A9s=20Ajna?= Date: Mon, 4 Dec 2023 13:01:18 -0800 Subject: [PATCH 1/2] Update pipeline_demofusion_sdxl.py --- examples/community/pipeline_demofusion_sdxl.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/community/pipeline_demofusion_sdxl.py b/examples/community/pipeline_demofusion_sdxl.py index 5a81320219a5..98508b7ff89c 100644 --- a/examples/community/pipeline_demofusion_sdxl.py +++ b/examples/community/pipeline_demofusion_sdxl.py @@ -36,7 +36,9 @@ if is_invisible_watermark_available(): - from .watermark import StableDiffusionXLWatermarker + from diffusers.pipelines.stable_diffusion_xl.watermark import ( + StableDiffusionXLWatermarker, + ) logger = logging.get_logger(__name__) # pylint: disable=invalid-name From c92eb885d19db1511d45c5dd027393c82389da8a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radam=C3=A9s=20Ajna?= Date: Mon, 4 Dec 2023 13:04:54 -0800 Subject: [PATCH 2/2] Update README.md --- examples/community/README.md | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/examples/community/README.md b/examples/community/README.md index 9fad6ecbf690..98780edeccf7 100755 --- a/examples/community/README.md +++ b/examples/community/README.md @@ -2870,10 +2870,14 @@ The original repo can be found at [repo](https://github.com/PRIS-CV/DemoFusion). - `show_image` (`bool`, defaults to False): Determine whether to show intermediate results during generation. ``` -from pipeline_demofusion_sdxl import DemoFusionSDXLPipeline +from diffusers import DiffusionPipeline -model_ckpt = "stabilityai/stable-diffusion-xl-base-1.0" -pipe = DemoFusionSDXLPipeline.from_pretrained(model_ckpt, torch_dtype=torch.float16) +pipe = DiffusionPipeline.from_pretrained( + "stabilityai/stable-diffusion-xl-base-1.0", + custom_pipeline="pipeline_demofusion_sdxl", + custom_revision="main", + torch_dtype=torch.float16, +) pipe = pipe.to("cuda") prompt = "Envision a portrait of an elderly woman, her face a canvas of time, framed by a headscarf with muted tones of rust and cream. Her eyes, blue like faded denim. Her attire, simple yet dignified."