(self, decoder, lm_head, final_logits_bias, config)
| 116 | """ |
| 117 | |
| 118 | def __init__(self, decoder, lm_head, final_logits_bias, config): |
| 119 | super().__init__() |
| 120 | self.decoder = decoder |
| 121 | self.lm_head = lm_head |
| 122 | self.bias = final_logits_bias |
| 123 | self.config = config |
| 124 | |
| 125 | @staticmethod |
| 126 | def _reorder_cache(past, beam_idx): |