noether.modeling.diffusion.factory

Attributes

AnyDiffusionScheduleConfig

Discriminated union of all built-in diffusion schedule configurations.

Functions

build_schedule(config)

Instantiate the right DiffusionSchedule for config.

Module Contents

noether.modeling.diffusion.factory.AnyDiffusionScheduleConfig

Discriminated union of all built-in diffusion schedule configurations.

Pydantic resolves the right variant by inspecting the kind field. Pair with build_schedule() to materialize the schedule object.

noether.modeling.diffusion.factory.build_schedule(config)

Instantiate the right DiffusionSchedule for config.

Parameters:

config (AnyDiffusionScheduleConfig) – Any variant of AnyDiffusionScheduleConfig.

Returns:

A DiffusionSchedule matching the variant’s kind.

Raises:

ValueError – If config is not a recognised schedule config.

Return type:

noether.modeling.diffusion.base.DiffusionSchedule