Skip to content

scheduling_minimax_h3

Classes

fastvideo.models.schedulers.scheduling_minimax_h3.MiniMaxH3Scheduler 

MiniMaxH3Scheduler(shift: float = 12.0)

Bases: SchedulerMixin, ConfigMixin

Rectified-flow scheduler using MiniMax-H3's clean-time convention.

Source code in fastvideo/models/schedulers/scheduling_minimax_h3.py 
23
24
25
26
27
28
29
30
31
32
@register_to_config
def __init__(self, shift: float = 12.0) -> None:
    if shift <= 0:
        raise ValueError(f"`shift` must be positive, got {shift}.")
    self.num_inference_steps: int | None = None
    self.sigmas: torch.Tensor | None = None
    self.timesteps: torch.Tensor | None = None
    self._shift = float(shift)
    self._step_index: int | None = None
    self._begin_index: int | None = None