fastvideo.v1.configs.models.dits.base

fastvideo.v1.configs.models.dits.base#

Module Contents#

Classes#

DiTArchConfig

DiTConfig

API#

class fastvideo.v1.configs.models.dits.base.DiTArchConfig[source]#

Bases: fastvideo.v1.configs.models.base.ArchConfig

exclude_lora_layers: list[str][source]#

‘field(…)’

hidden_size: int[source]#

0

lora_param_names_mapping: dict[source]#

‘field(…)’

num_attention_heads: int[source]#

0

num_channels_latents: int[source]#

0

param_names_mapping: dict[source]#

‘field(…)’

reverse_param_names_mapping: dict[source]#

‘field(…)’

class fastvideo.v1.configs.models.dits.base.DiTConfig[source]#

Bases: fastvideo.v1.configs.models.base.ModelConfig

static add_cli_args(parser: Any, prefix: str = 'dit-config') → Any[source]#

Add CLI arguments for DiTConfig fields

arch_config: fastvideo.v1.configs.models.dits.base.DiTArchConfig[source]#

‘field(…)’

prefix: str = <Multiline-String>[source]#
quant_config: fastvideo.v1.layers.quantization.QuantizationConfig | None[source]#

None

Contents