base ¶

Classes¶

fastvideo.models.encoders.base.TextEncoder ¶

TextEncoder(config: TextEncoderConfig)

Bases: Module, ABC, Generic[TextEncoderOutputT]

Base for native encoders with a model-specific forward output contract.

Source code in fastvideo/models/encoders/base.py

def __init__(self, config: TextEncoderConfig) -> None:
    super().__init__()
    self.config = config
    self._fsdp_shard_conditions = config._fsdp_shard_conditions
    self._stacked_params_mapping = config.arch_config.stacked_params_mapping
    if not self.supported_attention_backends:
        raise ValueError(f"Subclass {self.__class__.__name__} must define _supported_attention_backends")