diff --git a/src/axolotl/utils/collators/__init__.py b/src/axolotl/utils/collators/__init__.py index 649356e12..93502b67d 100644 --- a/src/axolotl/utils/collators/__init__.py +++ b/src/axolotl/utils/collators/__init__.py @@ -7,5 +7,4 @@ from .batching import ( # noqa: F401 PretrainingBatchSamplerDataCollatorForSeq2Seq, V2BatchSamplerDataCollatorForSeq2Seq, ) -from .kd import DataCollatorForKD # noqa: F401 from .mamba import MambaDataCollator # noqa: F401