fastvideo.configs.models.vaes.stepvideovae

`fastvideo.configs.models.vaes.stepvideovae`#

Module Contents#

Classes#

`StepVideoVAEArchConfig`
`StepVideoVAEConfig`

API#

class fastvideo.configs.models.vaes.stepvideovae.StepVideoVAEArchConfig[source]#

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

frame_len: int[source]#: 17

in_channels: int[source]#: 3

num_res_blocks: int[source]#: 2

out_channels: int[source]#: 3

scaling_factor: float[source]#: 1.0

spatial_compression_ratio: int[source]#: 16

temporal_compression_ratio: int[source]#: 8

version: int[source]#: 2

world_size: int[source]#: 1

z_channels: int[source]#: 64

class fastvideo.configs.models.vaes.stepvideovae.StepVideoVAEConfig[source]#

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

arch_config: fastvideo.configs.models.vaes.base.VAEArchConfig[source]#: ‘field(…)’

use_parallel_tiling: bool[source]#: False

use_temporal_scaling_frames: bool#: False

use_temporal_tiling: bool[source]#: False

use_tiling: bool[source]#: False