From 45392cce11c566a7df7aeeddf845dbd24a3b2311 Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Fri, 20 Jun 2025 07:38:21 +0200 Subject: [PATCH] update the description of StableDiffusionXLDenoiseLoopWrapper --- .../stable_diffusion_xl/denoise.py | 38 +++++++++++-------- 1 file changed, 23 insertions(+), 15 deletions(-) diff --git a/src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py b/src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py index 4d7ab12cf0..4485c17e97 100644 --- a/src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py +++ b/src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py @@ -687,9 +687,11 @@ class StableDiffusionXLDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper): def description(self) -> str: return ( "Denoise step that iteratively denoise the latents. " - "Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` " - "and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLDenoiseLoopBeforeDenoiser` and `StableDiffusionXLDenoiseLoopDenoiser`, " - "and finally `StableDiffusionXLDenoiseLoopAfterDenoiser` to update the latents." + "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n" + "and at each iteration, it runs blocks defined in `blocks` sequencially:\n" + " - `StableDiffusionXLDenoiseLoopBeforeDenoiser`\n" + " - `StableDiffusionXLDenoiseLoopDenoiser`\n" + " - `StableDiffusionXLDenoiseLoopAfterDenoiser`\n" ) # control_cond @@ -699,10 +701,12 @@ class StableDiffusionXLControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper @property def description(self) -> str: return ( - "Denoise step that iteratively denoise the latents with controlnet. " - "Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` " - "and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLDenoiseLoopBeforeDenoiser` and `StableDiffusionXLControlNetDenoiseLoopDenoiser`, " - "and finally `StableDiffusionXLDenoiseLoopAfterDenoiser` to update the latents." + "Denoise step that iteratively denoise the latents with controlnet. \n" + "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n" + "and at each iteration, it runs blocks defined in `blocks` sequencially:\n" + " - `StableDiffusionXLDenoiseLoopBeforeDenoiser`\n" + " - `StableDiffusionXLControlNetDenoiseLoopDenoiser`\n" + " - `StableDiffusionXLDenoiseLoopAfterDenoiser`\n" ) # mask @@ -712,10 +716,12 @@ class StableDiffusionXLInpaintDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper): @property def description(self) -> str: return ( - "Denoise step that iteratively denoise the latents(for inpainting task only). " - "Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` " - "and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser` and `StableDiffusionXLDenoiseLoopDenoiser`, " - "and finally `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser` to update the latents." + "Denoise step that iteratively denoise the latents(for inpainting task only). \n" + "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n" + "and at each iteration, it runs blocks defined in `blocks` sequencially:\n" + " - `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser`\n" + " - `StableDiffusionXLDenoiseLoopDenoiser`\n" + " - `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser`\n" ) # control_cond + mask class StableDiffusionXLInpaintControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper): @@ -724,10 +730,12 @@ class StableDiffusionXLInpaintControlNetDenoiseLoop(StableDiffusionXLDenoiseLoop @property def description(self) -> str: return ( - "Denoise step that iteratively denoise the latents(for inpainting task only) with controlnet. " - "Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` " - "and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser` and `StableDiffusionXLControlNetDenoiseLoopDenoiser`, " - "and finally `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser` to update the latents." + "Denoise step that iteratively denoise the latents(for inpainting task only) with controlnet. \n" + "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n" + "and at each iteration, it runs blocks defined in `blocks` sequencially:\n" + " - `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser`\n" + " - `StableDiffusionXLControlNetDenoiseLoopDenoiser`\n" + " - `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser`\n" )