Skip to content

enable flux pipeline compatible with unipc and dpm-solver #11908

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 9 commits into
base: main
Choose a base branch
from

Conversation

gameofdimension
Copy link
Contributor

@gameofdimension gameofdimension commented Jul 10, 2025

with this pr, we can use dpm/unipc together with flux pipeline by

pipe.scheduler = DPMSolverMultistepScheduler.from_config(
    pipe.scheduler.config, use_flow_sigmas=True, prediction_type="flow_prediction")

or

pipe.scheduler = UniPCMultistepScheduler.from_config(
    pipe.scheduler.config, use_flow_sigmas=True, prediction_type="flow_prediction")

What does this PR do?

Fixes #11907

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

have flux pipeline work with unipc/dpm schedulers
@gameofdimension gameofdimension changed the title enable flux pipeline compatible with unipc and dpm enable flux pipeline compatible with unipc and dpm-solver Jul 10, 2025
@yiyixuxu
Copy link
Collaborator

ohh thanks so much @gameofdimension
does it make sense to add use_dynamic_shifting config to dpm/unipc when use_flow_sigmas is enabled?

essentially, are we able to:

  1. not adding any changes to the pipeline
  2. add use_dynamic_shifting config the dpm/unipc scheduler, and accept the optional mu in their set_timesteps
  3. use this API at runtime to switch pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config,use_flow_sigma=True, prediction_type = "flow_prediction" )

@gameofdimension
Copy link
Contributor Author

Noted, I'll improve that

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

flux pipeline not work with DPMSolverMultistepScheduler and UniPCMultistepScheduler
2 participants