fastvideo.v1.configs.models.vaes.wanvae

`fastvideo.v1.configs.models.vaes.wanvae`#

Module Contents#

Classes#

`WanVAEArchConfig`
`WanVAEConfig`

API#

class fastvideo.v1.configs.models.vaes.wanvae.WanVAEArchConfig[source]#

Bases: fastvideo.v1.configs.models.vaes.base.VAEArchConfig

attn_scales: Tuple[float, ...][source]#: ()

base_dim: int[source]#: 96

dim_mult: Tuple[int, ...][source]#: (1, 2, 4, 4)

dropout: float[source]#: 0.0

latents_mean: Tuple[float, ...][source]#: ()

latents_std: Tuple[float, ...][source]#: (2.8184, 1.4541, 2.3275, 2.6558, 1.2196, 1.7708, 2.6052, 2.0743, 3.2687, 2.1526, 2.8652, 1.5579, 1.6…

num_res_blocks: int[source]#: 2

spatial_compression_ratio[source]#: 8

temperal_downsample: Tuple[bool, ...][source]#: (False, True, True)

temporal_compression_ratio[source]#: 4

z_dim: int[source]#: 16

class fastvideo.v1.configs.models.vaes.wanvae.WanVAEConfig[source]#

Bases: fastvideo.v1.configs.models.vaes.base.VAEConfig

arch_config: fastvideo.v1.configs.models.vaes.wanvae.WanVAEArchConfig[source]#: ‘field(…)’

use_feature_cache: bool[source]#: True

use_parallel_tiling: bool[source]#: False

use_temporal_tiling: bool[source]#: False

use_tiling: bool[source]#: False