fix import in block mapping

yiyixuxu · yiyixuxu · commit de6ab6b49d17 · 2025-05-20T18:07:58.000+02:00
diff --git a/src/diffusers/modular_pipelines/stable_diffusion_xl/modular_pipeline_block_mappings.py b/src/diffusers/modular_pipelines/stable_diffusion_xl/modular_pipeline_block_mappings.py
@@ -41,11 +41,11 @@
     StableDiffusionXLInpaintVaeEncoderStep,
     StableDiffusionXLIPAdapterStep
 )
-from .after_denoise import (
+from .decoders import (
     StableDiffusionXLDecodeStep,
-    StableDiffusionXLInpaintDecodeStep
+    StableDiffusionXLInpaintDecodeStep,
+    StableDiffusionXLAutoDecodeStep
 )
-from .after_denoise import StableDiffusionXLAutoDecodeStep
 
 
 # YiYi notes: comment out for now, work on this later

Original file line number	Diff line number	Diff line change
`@@ -41,11 +41,11 @@`
`41`	`41`	`StableDiffusionXLInpaintVaeEncoderStep,`
`42`	`42`	`StableDiffusionXLIPAdapterStep`
`43`	`43`	`)`
`44`		`-from .after_denoise import (`
	`44`	`+from .decoders import (`
`45`	`45`	`StableDiffusionXLDecodeStep,`
`46`		`- StableDiffusionXLInpaintDecodeStep`
	`46`	`+ StableDiffusionXLInpaintDecodeStep,`
	`47`	`+ StableDiffusionXLAutoDecodeStep`
`47`	`48`	`)`
`48`		`-from .after_denoise import StableDiffusionXLAutoDecodeStep`
`49`	`49`
`50`	`50`
`51`	`51`	`# YiYi notes: comment out for now, work on this later`