| 105 | |
| 106 | @config_class |
| 107 | class Config(BaseLayer.Config): |
| 108 | input_dim: Required[int] = REQUIRED |
| 109 | output_dim: Required[int] = REQUIRED |
| 110 | linear: InstantiableConfig = Linear.default_config() |
| 111 | activation: str = "nn.gelu" |
| 112 | norm: InstantiableConfig = LayerNorm.default_config() |
| 113 | |
| 114 | def __init__(self, cfg: Config, *, parent: Optional[Module]): |
| 115 | super().__init__(cfg, parent=parent) |
nothing calls this directly
no test coverage detected