huggingface / huggingface/diffusers

Request the UniPCMultistepScheduler to allow a customed timesteps

Open
#14,442 1 comment 0 reactions 0 assignees View on GitHub
feature-request schedulers
Dominant language
Python
Stars
34.5k
Forks
7.3k
Avg merge
3d 3h
Merged PRs (30d)
91

Description

**Is your feature request related to a problem? Please describe.**
Current UniPC Solver don't allow user set the timesteps by himself. Although it allow user set the sigmas, but that is for those diffusion model who allow the use_flow_sigmas to be true.

**Describe the solution you'd like.**
provide a set_timesteps function similar to the DPM Multistep Solver, which allow user set the timesteps by himself.

**Describe alternatives you've considered.**
I do think allowing users to set sigma even when the use_flow_sigmas is false is also an option, but that is not align with the previous solution in the DPM Multistep Solver

**Additional context.**
ODE solver for SD is still a popular research area, lots of algorithm, like LD3 from Nvidia, search a good timeschedule for the LDM.

Contributor guide

Open the contributing guide

Research direction

Start by locating UniPCMultistepScheduler and compare its timestep handling with the DPM Multistep Solver's set_timesteps function. Check how custom timesteps interact with use_flow_sigmas being false; done means users can supply their own timesteps through the scheduler with behavior covered by the relevant scheduler tests.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
62/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.