(self, prefix: str = '', recurse: bool = True)
| 21 | return self.model.load_state_dict(state_dict, strict=strict) |
| 22 | |
| 23 | def named_parameters(self, prefix: str = '', recurse: bool = True): |
| 24 | return self.model.named_parameters(prefix=prefix, recurse=recurse) |
| 25 | |
| 26 | def forward(self, input_ids, position_ids, attention_mask, target_ids=None, logit_mask=None, prompt_pos=None): |
| 27 | if target_ids == None: |