Skip to content

mmaudio

Classes

Functions:

fastvideo.configs.models.dits.mmaudio.get_mmaudio_transformer_config

get_mmaudio_transformer_config(variant: str) -> MMAudioTransformerConfig

Build the exact official architecture config for variant.

Source code in fastvideo/configs/models/dits/mmaudio.py
def get_mmaudio_transformer_config(variant: str) -> MMAudioTransformerConfig:
    """Build the exact official architecture config for ``variant``."""
    try:
        architecture = MMAUDIO_VARIANT_ARCHITECTURES[variant]
    except KeyError as exc:
        supported = ", ".join(MMAUDIO_VARIANT_ARCHITECTURES)
        raise ValueError(f"Unknown MMAudio variant {variant!r}; expected one of: {supported}") from exc
    return MMAudioTransformerConfig(arch_config=MMAudioArchConfig(**architecture), )