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

hidden_size: int[source]#

0

num_attention_heads: int[source]#

0

num_channels_latents: int[source]#

0

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: Optional[fastvideo.v1.layers.quantization.QuantizationConfig][source]#

None

Contents