ls_mlkit.diffuser.lie_group_diffuser module

Lie Group Diffuser

class ls_mlkit.diffuser.lie_group_diffuser.LieGroupDiffuser(config: LieGroupDiffuserConfig, time_scheduler: TimeScheduler, lie_group: LieGroup)

Bases: RiemannianManifoldDiffuser

Riemannian Manifold Diffuser

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

Bases: RiemannianManifoldDiffuserConfig

Lie Group Diffuser Config