Method build_layer

codegeex/paddle/codegeex_model.py:591–592

(layer_number)

        
        # Transformer layers.
        def build_layer(layer_number):
            return TransformerLayer(self.hidden_size, self.num_attention_heads, layer_number)

        self.layers = paddle.nn.LayerList(
            [build_layer(i + 1) for i in range(self.num_unique_layers)])