Initialize the weights.
(self, module: nn.Module)
| 671 | _no_split_modules = ["GLMBlock"] |
| 672 | |
| 673 | def _init_weights(self, module: nn.Module): |
| 674 | """Initialize the weights.""" |
| 675 | return |
| 676 | |
| 677 | def get_masks(self, input_ids, past_key_values, padding_mask=None): |
| 678 | batch_size, seq_length = input_ids.shape |
nothing calls this directly
no outgoing calls
no test coverage detected