fastvideo.v1.configs.models.vaes.base

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

Module Contents#

Classes#

`VAEArchConfig`
`VAEConfig`

API#

class fastvideo.v1.configs.models.vaes.base.VAEArchConfig[source]#

Bases: fastvideo.v1.configs.models.base.ArchConfig

scaling_factor: Union[float, torch.tensor][source]#: 0

spatial_compression_ratio: int[source]#: 8

temporal_compression_ratio: int[source]#: 4

class fastvideo.v1.configs.models.vaes.base.VAEConfig[source]#

Bases: fastvideo.v1.configs.models.base.ModelConfig

static add_cli_args(parser: Any, prefix: str = 'vae-config') → Any[source]#: Add CLI arguments for VAEConfig fields

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

blend_num_frames: int[source]#: 0

load_decoder: bool[source]#: True

load_encoder: bool[source]#: True

tile_sample_min_height: int[source]#: 256

tile_sample_min_num_frames: int[source]#: 16

tile_sample_min_width: int[source]#: 256

tile_sample_stride_height: int[source]#: 192

tile_sample_stride_num_frames: int[source]#: 12

tile_sample_stride_width: int[source]#: 192

use_parallel_tiling: bool[source]#: True

use_temporal_scaling_frames: bool[source]#: True

use_temporal_tiling: bool[source]#: True

use_tiling: bool[source]#: True