Using euler scheduler in fluxfill

[luna313]

Describe the bug

I am using the customfluxfill function and want to use the Euler scheduler (EulerAncestralDiscreteScheduler) in my code. However, I am encountering the following error:

Reproduction

from diffusers.schedulers import (
DPMSolverMultistepScheduler,
    EulerAncestralDiscreteScheduler,
)

SAMPLER_MAP = {
    "DPM++ SDE Karras": lambda config: DPMSolverMultistepScheduler.from_config(
        config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++"
    ),
    "DPM++ 2M Karras": lambda config: DPMSolverMultistepScheduler.from_config(
        config, use_karras_sigmas=True
    ),
    "DPM++ 2M SDE": lambda config: DPMSolverMultistepScheduler.from_config(
        config, algorithm_type="sde-dpmsolver++"
    ),
    "DPM++ 2M SDE Karras": lambda config: DPMSolverMultistepScheduler.from_config(
        config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++"
    ),
    "Euler a": lambda config: EulerAncestralDiscreteScheduler.from_config(
        config,
    ),
}

def model_fn() -> FluxFillPipeline:
    pipe = FluxFillPipeline.from_pretrained("Planningo/flux-inpaint", torch_dtype=torch.bfloat16).to("cuda")
    # Apply the Euler scheduler
    pipe.scheduler = SAMPLER_MAP["Euler a"](pipe.scheduler.config)
    return pipe

Logs

ValueError: The current scheduler class <class 'diffusers.schedulers.scheduling_euler_ancestral_discrete.EulerAncestralDiscreteScheduler'>'s `set_timesteps` does not support custom sigmas schedules. Please check whether you are using the correct scheduler.

System Info

latest(github version diffusers), python3.10, ubuntu with nvidia gpu

Who can help?

@yiyixuxu

[hlky]

Hi @luna313! We are working on supporting more schedulers for Flow Match models like Flux, we support FlowMatchEulerDiscreteScheduler for now.

[luna313]

@hlky Thank you for your response! I hope that the feature will be supported soon.