MCPcopy Create free account
hub / github.com/THUDM/LongWriter / _get_layer

Method _get_layer

train/patch/modeling_chatglm.py:515–516  ·  view source on GitHub ↗
(self, layer_number)

Source from the content-addressed store, hash-verified

513 self.gradient_checkpointing = False
514
515 def _get_layer(self, layer_number):
516 return self.layers[layer_number]
517
518 def forward(
519 self, hidden_states, attention_mask, rotary_pos_emb, kv_caches=None,

Callers 1

forwardMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected