Skip to main content

Python class

DenoisingCacheConfig

DenoisingCacheConfig

class max.pipelines.modeling.base.DenoisingCacheConfig(*, config_file=None, section_name=None, first_block_caching=False, taylorseer=False, taylorseer_cache_interval=None, taylorseer_warmup_steps=None, taylorseer_max_order=None)

source

Bases: ConfigFileModel

Pipeline-level cache configuration for diffusion model denoising.

Controls First-Block Cache (step cache) and TaylorSeer optimizations that skip redundant transformer passes during the denoising loop.

Parameters:

  • config_file (str | None)
  • section_name (str | None)
  • first_block_caching (bool)
  • taylorseer (bool)
  • taylorseer_cache_interval (int | None)
  • taylorseer_warmup_steps (int | None)
  • taylorseer_max_order (int | None)

config_file

config_file: str | None

source

Path to the configuration file.

first_block_caching

first_block_caching: bool

source

model_config

model_config: ClassVar[ConfigDict] = {'extra': 'forbid', 'frozen': False, 'strict': False}

source

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

section_name

section_name: str | None

source

Optional section name for comprehensive/multi-section config files.

If not provided, values are loaded from the YAML top-level (treating the file as an “individual config” file).

taylorseer

taylorseer: bool

source

taylorseer_cache_interval

taylorseer_cache_interval: int | None

source

taylorseer_max_order

taylorseer_max_order: int | None

source

taylorseer_warmup_steps

taylorseer_warmup_steps: int | None

source