ls_mlkit.diffusion.base_diffuser module

class ls_mlkit.diffusion.base_diffuser.BaseDiffuser(config: BaseDiffuserConfig, time_scheduler: DiffusionTimeScheduler)

Bases: BaseGenerativeModel

abstract method:

abstractmethod forward_process(x_0: Tensor, t_a: Tensor, t_b: Tensor, mask: Tensor, is_continuous_time: bool = True, *args: list[Any], **kwargs: dict[Any, Any]) → dict

class ls_mlkit.diffusion.base_diffuser.BaseDiffuserConfig(ndim_micro_shape: int, n_discretization_steps: int, n_inference_steps: int = None, *args: list[Any], **kwargs: dict[Any, Any])

Bases: BaseGenerativeModelConfig