Class Config

model.py:19–26

@si_module
class LatentQuantizer(nn.Module):
    class Config:
        compressor_config: Optional[FSQ.Config] = None

        dim: Optional[int] = None
        ff_dim: Optional[int] = None
        input_dim: int = None

        from_pretrained: Optional[Tuple[str, str]] = None

    def __init__(self, c: Config):
        super().__init__()