add

Author: stevhliu

src/diffusers/schedulers/scheduling_dpmsolver_multistep.py

@@ -177,6 +177,7 @@ class DPMSolverMultistepScheduler(SchedulerMixin, ConfigMixin):
         use_flow_sigmas (`bool`, *optional*, defaults to `False`):
             Whether to use flow sigmas for step sizes in the noise schedule during the sampling process.
         flow_shift (`float`, *optional*, defaults to 1.0):
+            The shift value for the timestep schedule for flow matching.
         final_sigmas_type (`str`, defaults to `"zero"`):
             The final `sigma` value for the noise schedule during the sampling process. If `"sigma_min"`, the final
             sigma is the same as the last sigma in the training schedule. If `zero`, the final sigma is set to 0.

src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py

@@ -71,6 +71,7 @@ class FlowMatchEulerDiscreteScheduler(SchedulerMixin, ConfigMixin):
         invert_sigmas (`bool`, defaults to False):
             Whether to invert the sigmas.
         shift_terminal (`float`, defaults to None):
+            The end value of the shifted timestep schedule.
         use_karras_sigmas (`bool`, defaults to False):
             Whether to use Karras sigmas for step sizes in the noise schedule during sampling.
         use_exponential_sigmas (`bool`, defaults to False):