chore: lint

2024-12-19 02:04:25 -05:00
parent e2fae47114
commit ba6eacd167
1 changed files with 1 additions and 0 deletions
--- a/src/axolotl/utils/collators/kd.py
+++ b/src/axolotl/utils/collators/kd.py
@@ -23,6 +23,7 @@ class DataCollatorForKD(DataCollatorForSeq2Seq):
    target_logprobs. It also creates a teacher_mask to indicate which entries are valid.
    """

+    # pylint: disable=duplicate-code
    tokenizer: PreTrainedTokenizerBase
    model: Optional[Any] = None
    padding: Union[bool, str, PaddingStrategy] = True