(
self,
hidden_size,
num_layers,
num_attention_heads,
padded_vocab_size,
max_position_embeddings,
)
| 948 | """CodeGeeX: A Multilingual Code Generation Model.""" |
| 949 | |
| 950 | def __init__( |
| 951 | self, |
| 952 | hidden_size, |
| 953 | num_layers, |
| 954 | num_attention_heads, |
| 955 | padded_vocab_size, |
| 956 | max_position_embeddings, |
| 957 | ): |
| 958 | super(CodeGeeXModel, self).__init__() |
| 959 | |
| 960 | self.language_model = TransformerLanguageModel(hidden_size, |
| 961 | num_layers, |
| 962 | num_attention_heads, |
| 963 | padded_vocab_size, |
| 964 | max_position_embeddings) |
| 965 | self._language_model_key = "language_model" |
| 966 | |
| 967 | def forward( |
| 968 | self, |
nothing calls this directly
no test coverage detected