fastvideo.v1.configs.models.dits.stepvideo

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

Module Contents#

Classes#

StepVideoArchConfig

StepVideoConfig

Functions#

is_blocks

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: Optional[str][source]#

‘torch’

caption_channels: Optional[Union[int, List[int], Tuple[int, ...]]][source]#

‘field(…)’

dropout: float[source]#

0.0

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: Optional[int][source]#

64

patch_size: int[source]#

1

use_additional_conditions: Optional[bool][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’

fastvideo.v1.configs.models.dits.stepvideo.is_blocks(n: str, m) → bool[source]#
Contents