AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

update

Browse Source
This commit is contained in:
Dhruv Nair
2025-08-08 18:22:44 +02:00
parent 919ee1aee3
commit 085e9cba36
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/diffusers/modular_pipelines/wan/modular_blocks.py
View File

@@ -13,7 +13,7 @@
# limitations under the License.
from ...utils import logging
from ..modular_pipeline import AutoModularPipelineBlockss, SequentialModularPipelineBlockss
from ..modular_pipeline import AutoPipelineBlocks, SequentialPipelineBlocks
from ..modular_pipeline_utils import InsertableDict
from .before_denoise import (
WanInputStep,
@@ -29,7 +29,7 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name
# before_denoise: text2vid
class WanBeforeDenoiseStep(SequentialModularPipelineBlockss):
class WanBeforeDenoiseStep(SequentialPipelineBlocks):
block_classes = [
WanInputStep,
WanSetTimestepsStep,
@@ -49,7 +49,7 @@ class WanBeforeDenoiseStep(SequentialModularPipelineBlockss):
# before_denoise: all task (text2vid,)
class WanAutoBeforeDenoiseStep(AutoModularPipelineBlockss):
class WanAutoBeforeDenoiseStep(AutoPipelineBlocks):
block_classes = [
WanBeforeDenoiseStep,
]
@@ -66,7 +66,7 @@ class WanAutoBeforeDenoiseStep(AutoModularPipelineBlockss):
# denoise: text2vid
class WanAutoDenoiseStep(AutoModularPipelineBlockss):
class WanAutoDenoiseStep(AutoPipelineBlocks):
block_classes = [
WanDenoiseStep,
]
@@ -83,7 +83,7 @@ class WanAutoDenoiseStep(AutoModularPipelineBlockss):
# decode: all task (text2img, img2img, inpainting)
class WanAutoDecodeStep(AutoModularPipelineBlockss):
class WanAutoDecodeStep(AutoPipelineBlocks):
block_classes = [WanDecodeStep]
block_names = ["non-inpaint"]
block_trigger_inputs = [None]
@@ -94,7 +94,7 @@ class WanAutoDecodeStep(AutoModularPipelineBlockss):
# text2vid
class WanAutoBlocks(SequentialModularPipelineBlockss):
class WanAutoBlocks(SequentialPipelineBlocks):
block_classes = [
WanTextEncoderStep,
WanAutoBeforeDenoiseStep,