ls_mlkit.diffusion.lie_group_diffuser module

Lie Group Diffuser

class ls_mlkit.diffusion.lie_group_diffuser.LieGroupDiffuser(config: LieGroupDiffuserConfig, time_scheduler: DiffusionTimeScheduler, lie_group: LieGroup)

Bases: RiemannianManifoldDiffuser

class ls_mlkit.diffusion.lie_group_diffuser.LieGroupDiffuserConfig(ndim_micro_shape: int, n_discretization_steps: int, n_inference_steps: int, *args: list[Any], **kwargs: dict[Any, Any])

Bases: RiemannianManifoldDiffuserConfig