fastvideo.v1.configs.models.dits.stepvideo

`fastvideo.v1.configs.models.dits.stepvideo`#

Module Contents#

Classes#

`StepVideoArchConfig`
`StepVideoConfig`

API#

class fastvideo.v1.configs.models.dits.stepvideo.StepVideoArchConfig[source]#

Bases: fastvideo.v1.configs.models.dits.base.DiTArchConfig

attention_head_dim: int[source]#: 128

attention_type: str | None[source]#: ‘torch’

caption_channels: int | list[int] | tuple[int, ...] | None[source]#: ‘field(…)’

dropout: float[source]#: 0.0

exclude_lora_layers: list[str][source]#: ‘field(…)’

in_channels: int[source]#: 64

norm_elementwise_affine: bool[source]#: False

norm_eps: float[source]#: 1e-06

norm_type: str[source]#: ‘ada_norm_single’

num_attention_heads: int[source]#: 48

num_layers: int[source]#: 48

out_channels: int | None[source]#: 64

param_names_mapping: dict[source]#: ‘field(…)’

patch_size: int[source]#: 1

use_additional_conditions: bool | None[source]#: False

class fastvideo.v1.configs.models.dits.stepvideo.StepVideoConfig[source]#

Bases: fastvideo.v1.configs.models.dits.base.DiTConfig

arch_config: fastvideo.v1.configs.models.dits.base.DiTArchConfig[source]#: ‘field(…)’

prefix: str#: ‘StepVideo’