use unipc instead of flow match

Author: a-r-r-o-w

src/diffusers/modular_pipelines/wan/before_denoise.py

@@ -17,7 +17,7 @@
 
 import torch
 
-from ...schedulers import FlowMatchEulerDiscreteScheduler
+from ...schedulers import UniPCMultistepScheduler
 from ...utils import logging
 from ...utils.torch_utils import randn_tensor
 from ..modular_pipeline import PipelineBlock, PipelineState
@@ -200,7 +200,7 @@ class WanSetTimestepsStep(PipelineBlock):
     @property
     def expected_components(self) -> List[ComponentSpec]:
         return [
-            ComponentSpec("scheduler", FlowMatchEulerDiscreteScheduler),
+            ComponentSpec("scheduler", UniPCMultistepScheduler),
         ]
 
     @property

src/diffusers/modular_pipelines/wan/denoise.py

@@ -19,7 +19,7 @@
 from ...configuration_utils import FrozenDict
 from ...guiders import ClassifierFreeGuidance
 from ...models import WanTransformer3DModel
-from ...schedulers import FlowMatchEulerDiscreteScheduler
+from ...schedulers import UniPCMultistepScheduler
 from ...utils import logging
 from ..modular_pipeline import (
     BlockState,
@@ -40,7 +40,7 @@ class WanLoopBeforeDenoiser(PipelineBlock):
     @property
     def expected_components(self) -> List[ComponentSpec]:
         return [
-            ComponentSpec("scheduler", FlowMatchEulerDiscreteScheduler),
+            ComponentSpec("scheduler", UniPCMultistepScheduler),
         ]
 
     @property
@@ -165,7 +165,7 @@ class WanLoopAfterDenoiser(PipelineBlock):
     @property
     def expected_components(self) -> List[ComponentSpec]:
         return [
-            ComponentSpec("scheduler", FlowMatchEulerDiscreteScheduler),
+            ComponentSpec("scheduler", UniPCMultistepScheduler),
         ]
 
     @property
@@ -227,7 +227,7 @@ def loop_expected_components(self) -> List[ComponentSpec]:
                 config=FrozenDict({"guidance_scale": 5.0}),
                 default_creation_method="from_config",
             ),
-            ComponentSpec("scheduler", FlowMatchEulerDiscreteScheduler),
+            ComponentSpec("scheduler", UniPCMultistepScheduler),
             ComponentSpec("transformer", WanTransformer3DModel),
         ]
 

src/diffusers/modular_pipelines/wan/encoders.py

@@ -107,6 +107,7 @@ def check_inputs(block_state):
         ):
             raise ValueError(f"`prompt` has to be of type `str` or `list` but is {type(block_state.prompt)}")
 
+    @staticmethod
     def _get_t5_prompt_embeds(
         components,
         prompt: Union[str, List[str]],