ls_mlkit.diffuser.manifold_diffuser module

Riemannian Manifold Diffuser

class ls_mlkit.diffuser.manifold_diffuser.RiemannianManifoldDiffuser(config: RiemannianManifoldDiffuserConfig, time_scheduler: TimeScheduler, riemannian_manifold: RiemannianManifold)

Bases: BaseDiffuser

Riemannian Manifold Diffuser

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

Bases: BaseDiffuserConfig

Riemannian Manifold Diffuser Config