fastvideo.v1.configs.models.encoders.base

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

Module Contents#

Classes#

`BaseEncoderOutput`
`EncoderArchConfig`
`EncoderConfig`
`ImageEncoderArchConfig`
`ImageEncoderConfig`
`TextEncoderArchConfig`
`TextEncoderConfig`

API#

class fastvideo.v1.configs.models.encoders.base.BaseEncoderOutput[source]#

attention_mask: Optional[torch.Tensor][source]#: None

attentions: Optional[Tuple[torch.FloatTensor, ...]][source]#: None

hidden_states: Optional[Tuple[torch.FloatTensor, ...]][source]#: None

last_hidden_state: Optional[torch.FloatTensor][source]#: None

pooler_output: Optional[torch.FloatTensor][source]#: None

class fastvideo.v1.configs.models.encoders.base.EncoderArchConfig[source]#

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

architectures: List[str][source]#: ‘field(…)’

output_hidden_states: bool[source]#: False

use_return_dict: bool[source]#: True

class fastvideo.v1.configs.models.encoders.base.EncoderConfig[source]#

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

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

lora_config: Optional[Any][source]#: None

prefix: str = <Multiline-String>[source]#

quant_config: Optional[fastvideo.v1.layers.quantization.QuantizationConfig][source]#: None

class fastvideo.v1.configs.models.encoders.base.ImageEncoderArchConfig[source]#: Bases: fastvideo.v1.configs.models.encoders.base.EncoderArchConfig

class fastvideo.v1.configs.models.encoders.base.ImageEncoderConfig[source]#

Bases: fastvideo.v1.configs.models.encoders.base.EncoderConfig

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

class fastvideo.v1.configs.models.encoders.base.TextEncoderArchConfig[source]#

Bases: fastvideo.v1.configs.models.encoders.base.EncoderArchConfig

decoder_start_token_id: int[source]#: 0

eos_token_id: int[source]#: 0

hidden_size: int[source]#: 0

hidden_state_skip_layer: int[source]#: 0

num_attention_heads: int[source]#: 0

num_hidden_layers: int[source]#: 0

output_past: bool[source]#: True

pad_token_id: int[source]#: 0

scalable_attention: bool[source]#: True

text_len: int[source]#: 0

tie_word_embeddings: bool[source]#: False

tokenizer_kwargs: Dict[str, Any][source]#: ‘field(…)’

vocab_size: int[source]#: 0

class fastvideo.v1.configs.models.encoders.base.TextEncoderConfig[source]#