Sgm uniform scheduler for SDXL-Lightning models #15325
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
Based on the example code/workflow from https://huggingface.co/ByteDance/SDXL-Lightning
I implement the sgm_uniform scheduler for SDXL-lightning series models (also support the SDXL Lightning LoRA models)
I create a new file for this custom scheduler since I guess we will have more and more custom scheduler in the future.
I don't know if we should make some extension API on it. But I will keep it clean at first.
SDXL Lightning Examples (4step Euler):
SDXL Lightning LoRA + KXL Delta Examples (4step Euler):
Checklist: