Skip to content

Commit b3f7212

Browse files
committed
change the FlowMatchEulerDiscreteScheduler to default DPMSolverMultistepScheduler with flow matching scheduler.
1 parent 98b778e commit b3f7212

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/sana/pipeline_sana.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from ...callbacks import MultiPipelineCallbacks, PipelineCallback
2525
from ...image_processor import PixArtImageProcessor
2626
from ...models import AutoencoderDC, SanaTransformer2DModel
27-
from ...schedulers import FlowMatchEulerDiscreteScheduler
27+
from ...schedulers import DPMSolverMultistepScheduler
2828
from ...utils import (
2929
BACKENDS_MAPPING,
3030
is_bs4_available,
@@ -137,7 +137,7 @@ def __init__(
137137
text_encoder: AutoModelForCausalLM,
138138
vae: AutoencoderDC,
139139
transformer: SanaTransformer2DModel,
140-
scheduler: FlowMatchEulerDiscreteScheduler,
140+
scheduler: DPMSolverMultistepScheduler,
141141
):
142142
super().__init__()
143143

0 commit comments

Comments
 (0)