(self, input_ids, **kwargs)
| 109 | freeze_params(d.embed_tokens) |
| 110 | |
| 111 | def forward(self, input_ids, **kwargs): |
| 112 | return self.model(input_ids, **kwargs) |
| 113 | |
| 114 | def ids_to_clean_text(self, generated_ids: List[int]): |
| 115 | gen_text = self.tokenizer.batch_decode( |