(self, config: FlexBertConfig)
| 621 | """ |
| 622 | |
| 623 | def __init__(self, config: FlexBertConfig): |
| 624 | super().__init__() |
| 625 | self.layers = nn.ModuleList([get_bert_layer(config, layer_id=i) for i in range(config.num_hidden_layers)]) |
| 626 | self.num_attention_heads = config.num_attention_heads |
| 627 | |
| 628 | def forward( |
| 629 | self, |
nothing calls this directly
no test coverage detected