diff --git a/src/axolotl/utils/collators/kd.py b/src/axolotl/utils/collators/kd.py
index 256dbe7a8..155840912 100644
--- 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