(
self,
hidden_size,
num_layers,
num_attention_heads,
padded_vocab_size,
max_position_embeddings,
)
| 1040 | """CodeGeeX: A Multilingual Code Generation Model.""" |
| 1041 | |
| 1042 | def __init__( |
| 1043 | self, |
| 1044 | hidden_size, |
| 1045 | num_layers, |
| 1046 | num_attention_heads, |
| 1047 | padded_vocab_size, |
| 1048 | max_position_embeddings, |
| 1049 | ): |
| 1050 | super(CodeGeeXModel, self).__init__() |
| 1051 | |
| 1052 | self.language_model = TransformerLanguageModel(hidden_size, |
| 1053 | num_layers, |
| 1054 | num_attention_heads, |
| 1055 | padded_vocab_size, |
| 1056 | max_position_embeddings) |
| 1057 | self._language_model_key = "language_model" |
| 1058 | |
| 1059 | def forward( |
| 1060 | self, |
nothing calls this directly
no test coverage detected