Class Config

model.py:201–212

@si_module
class HertzDevModel(nn.Module):
    class Config:
        dim: int
        vocab_size: int
        stack_config: Optional[Stack.Config] = None
        latent_size: int = 32

        split: bool = True

        quantizer_config: Optional[LatentQuantizer.Config] = None
        resynthesizer_config: Optional[TransformerVAE.Config] = None

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

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