Initialize the weights.
(self, module: nn.Module)
| 582 | _no_split_modules = ["GLMBlock"] |
| 583 | |
| 584 | def _init_weights(self, module: nn.Module): |
| 585 | """Initialize the weights.""" |
| 586 | return |
| 587 | |
| 588 | def get_masks(self, input_ids, past_key_values, padding_mask=None): |
| 589 | batch_size, seq_length = input_ids.shape |
nothing calls this directly
no outgoing calls
no test coverage detected