1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
This commit is contained in:
Dhruv Nair
2025-08-08 18:52:55 +02:00
parent 085e9cba36
commit 6c85fcd899

View File

@@ -23,7 +23,7 @@ from ...schedulers import UniPCMultistepScheduler
from ...utils import logging
from ..modular_pipeline import (
BlockState,
LoopSequentialModularPipelineBlockss,
LoopSequentialPipelineBlocks,
ModularPipelineBlocks,
PipelineState,
)
@@ -53,7 +53,7 @@ class WanLoopDenoiser(ModularPipelineBlocks):
def description(self) -> str:
return (
"Step within the denoising loop that denoise the latents with guidance. "
"This block should be used to compose the `sub_blocks` attribute of a `LoopSequentialModularPipelineBlockss` "
"This block should be used to compose the `sub_blocks` attribute of a `LoopSequentialPipelineBlocks` "
"object (e.g. `WanDenoiseLoopWrapper`)"
)
@@ -145,7 +145,7 @@ class WanLoopAfterDenoiser(ModularPipelineBlocks):
def description(self) -> str:
return (
"step within the denoising loop that update the latents. "
"This block should be used to compose the `sub_blocks` attribute of a `LoopSequentialModularPipelineBlockss` "
"This block should be used to compose the `sub_blocks` attribute of a `LoopSequentialPipelineBlocks` "
"object (e.g. `WanDenoiseLoopWrapper`)"
)
@@ -181,7 +181,7 @@ class WanLoopAfterDenoiser(ModularPipelineBlocks):
return components, block_state
class WanDenoiseLoopWrapper(LoopSequentialModularPipelineBlockss):
class WanDenoiseLoopWrapper(LoopSequentialPipelineBlocks):
model_name = "wan"
@property