Update modular_blocks.py
Browse files- modular_blocks.py +5 -6
modular_blocks.py
CHANGED
|
@@ -1,12 +1,11 @@
|
|
| 1 |
from diffusers.modular_pipelines import SequentialPipelineBlocks
|
| 2 |
from diffusers.modular_pipelines.modular_pipeline_utils import InsertableDict
|
| 3 |
-
from diffusers.modular_pipelines.wan.decoders import WanDecodeStep
|
| 4 |
|
| 5 |
-
from before_denoise import ChronoEditPrepareLatentStep, ChronoEditSetTimestepsStep
|
| 6 |
-
from denoise import ChronoEditDenoiseStep
|
| 7 |
-
from inputs import ChronoEditImageInputStep, ChronoEditInputStep
|
| 8 |
-
from encoders import ChronoEditImageEncoderStep, ChronoEditProcessImageStep, ChronoEditTextEncoderStep
|
| 9 |
-
from decoders import ChronoEditDecodeStep
|
| 10 |
|
| 11 |
|
| 12 |
class ChronoEditBeforeDenoiseStep(SequentialPipelineBlocks):
|
|
|
|
| 1 |
from diffusers.modular_pipelines import SequentialPipelineBlocks
|
| 2 |
from diffusers.modular_pipelines.modular_pipeline_utils import InsertableDict
|
|
|
|
| 3 |
|
| 4 |
+
from .before_denoise import ChronoEditPrepareLatentStep, ChronoEditSetTimestepsStep
|
| 5 |
+
from .denoise import ChronoEditDenoiseStep
|
| 6 |
+
from .inputs import ChronoEditImageInputStep, ChronoEditInputStep
|
| 7 |
+
from .encoders import ChronoEditImageEncoderStep, ChronoEditProcessImageStep, ChronoEditTextEncoderStep
|
| 8 |
+
from .decoders import ChronoEditDecodeStep
|
| 9 |
|
| 10 |
|
| 11 |
class ChronoEditBeforeDenoiseStep(SequentialPipelineBlocks):
|