Skip to content

Commit 45392cc

Browse files
committed
update the description of StableDiffusionXLDenoiseLoopWrapper
1 parent 8913d59 commit 45392cc

File tree

1 file changed

+23
-15
lines changed
  • src/diffusers/modular_pipelines/stable_diffusion_xl

1 file changed

+23
-15
lines changed

src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py

Lines changed: 23 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -687,9 +687,11 @@ class StableDiffusionXLDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
687687
def description(self) -> str:
688688
return (
689689
"Denoise step that iteratively denoise the latents. "
690-
"Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` "
691-
"and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLDenoiseLoopBeforeDenoiser` and `StableDiffusionXLDenoiseLoopDenoiser`, "
692-
"and finally `StableDiffusionXLDenoiseLoopAfterDenoiser` to update the latents."
690+
"Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
691+
"and at each iteration, it runs blocks defined in `blocks` sequencially:\n"
692+
" - `StableDiffusionXLDenoiseLoopBeforeDenoiser`\n"
693+
" - `StableDiffusionXLDenoiseLoopDenoiser`\n"
694+
" - `StableDiffusionXLDenoiseLoopAfterDenoiser`\n"
693695
)
694696

695697
# control_cond
@@ -699,10 +701,12 @@ class StableDiffusionXLControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper
699701
@property
700702
def description(self) -> str:
701703
return (
702-
"Denoise step that iteratively denoise the latents with controlnet. "
703-
"Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` "
704-
"and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLDenoiseLoopBeforeDenoiser` and `StableDiffusionXLControlNetDenoiseLoopDenoiser`, "
705-
"and finally `StableDiffusionXLDenoiseLoopAfterDenoiser` to update the latents."
704+
"Denoise step that iteratively denoise the latents with controlnet. \n"
705+
"Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
706+
"and at each iteration, it runs blocks defined in `blocks` sequencially:\n"
707+
" - `StableDiffusionXLDenoiseLoopBeforeDenoiser`\n"
708+
" - `StableDiffusionXLControlNetDenoiseLoopDenoiser`\n"
709+
" - `StableDiffusionXLDenoiseLoopAfterDenoiser`\n"
706710
)
707711

708712
# mask
@@ -712,10 +716,12 @@ class StableDiffusionXLInpaintDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
712716
@property
713717
def description(self) -> str:
714718
return (
715-
"Denoise step that iteratively denoise the latents(for inpainting task only). "
716-
"Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` "
717-
"and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser` and `StableDiffusionXLDenoiseLoopDenoiser`, "
718-
"and finally `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser` to update the latents."
719+
"Denoise step that iteratively denoise the latents(for inpainting task only). \n"
720+
"Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
721+
"and at each iteration, it runs blocks defined in `blocks` sequencially:\n"
722+
" - `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser`\n"
723+
" - `StableDiffusionXLDenoiseLoopDenoiser`\n"
724+
" - `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser`\n"
719725
)
720726
# control_cond + mask
721727
class StableDiffusionXLInpaintControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
@@ -724,10 +730,12 @@ class StableDiffusionXLInpaintControlNetDenoiseLoop(StableDiffusionXLDenoiseLoop
724730
@property
725731
def description(self) -> str:
726732
return (
727-
"Denoise step that iteratively denoise the latents(for inpainting task only) with controlnet. "
728-
"Its loop logic is defined in parent class `StableDiffusionXLDenoiseLoopWrapper` "
729-
"and at each iteration, it runs blocks defined in `blocks` sequencially, i.e. `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser` and `StableDiffusionXLControlNetDenoiseLoopDenoiser`, "
730-
"and finally `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser` to update the latents."
733+
"Denoise step that iteratively denoise the latents(for inpainting task only) with controlnet. \n"
734+
"Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
735+
"and at each iteration, it runs blocks defined in `blocks` sequencially:\n"
736+
" - `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser`\n"
737+
" - `StableDiffusionXLControlNetDenoiseLoopDenoiser`\n"
738+
" - `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser`\n"
731739
)
732740

733741

0 commit comments

Comments
 (0)