(self, *inputs, **kwargs)
| 570 | _skip_keys_device_placement = "past_key_values" |
| 571 | |
| 572 | def __init__(self, *inputs, **kwargs): |
| 573 | super().__init__(*inputs, **kwargs) |
| 574 | |
| 575 | def _init_weights(self, module: nn.Module): |
| 576 | """Initialize the weights.""" |