ls_mlkit.diffusion.manifold_diffuser module

Riemannian Manifold Diffuser

class ls_mlkit.diffusion.manifold_diffuser.RiemannianManifoldDiffuser(config: RiemannianManifoldDiffuserConfig, time_scheduler: DiffusionTimeScheduler, riemannian_manifold: RiemannianManifold)

Bases: BaseDiffuser

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

Bases: BaseDiffuserConfig