ls_mlkit.flow_matching.time_scheduler module

class ls_mlkit.flow_matching.time_scheduler.FlowMatchingTimeScheduler(continuous_time_start: float = 0.0, continuous_time_end: float = 1.0, num_train_timesteps: int = 1000, num_inference_steps: int = None)[source]

Bases: BaseTimeScheduler

initialize_timesteps_schedule() → None[source]

initialize timesteps for sampling.