fastvideo.v1.configs.models.dits.wanvideo

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

Module Contents#

Classes#

`WanVideoArchConfig`
`WanVideoConfig`

Functions#

API#

class fastvideo.v1.configs.models.dits.wanvideo.WanVideoArchConfig[source]#

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

added_kv_proj_dim: Optional[int][source]#: None

attention_head_dim: int[source]#: 128

cross_attn_norm: bool[source]#: True

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

ffn_dim: int[source]#: 13824

freq_dim: int[source]#: 256

image_dim: Optional[int][source]#: None

in_channels: int[source]#: 16

num_attention_heads: int[source]#: 40

num_layers: int[source]#: 40

out_channels: int[source]#: 16

patch_size: Tuple[int, int, int][source]#: (1, 2, 2)

qk_norm: str[source]#: ‘rms_norm_across_heads’

rope_max_seq_len: int[source]#: 1024

text_dim: int[source]#: 4096

text_len[source]#: 512

class fastvideo.v1.configs.models.dits.wanvideo.WanVideoConfig[source]#

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

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

prefix: str#: ‘Wan’

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