ls_mlkit.diffusion.time_scheduler module

Time Scheduler for Diffusion

class ls_mlkit.diffusion.time_scheduler.DiffusionTimeScheduler(continuous_time_start: float = 0.0, continuous_time_end: float = 1.0, num_train_timesteps: int = 1000, num_inference_steps: int = None, idx_start: int = 0)[source]

Bases: BaseTimeScheduler

initialize_timesteps_schedule() → None[source]

Initialize timesteps schedule for sampling/inference.