Improve docstrings and type hints in scheduling_deis_multistep.py - #12796

Merged

stevhliu merged 2 commits into

huggingface:mainfrom

delmalih:improve-docstring-scheduling-deis-multistep

Dec 5, 2025

Commits on Dec 5, 2025

feat: Add `flow_prediction` to `prediction_type`, introduce `use_flow_sigmas`, `flow_shift`, `use_dynamic_shifting`, and `time_shift_type` parameters, and refine type hints for various arguments.
delmalih
committed
4ea2ab0
style: reformat argument wrapping in `_convert_to_beta` and `index_for_timestep` method signatures.
delmalih
committed
337d843