remove moved class from import
This commit is contained in:
@@ -7,5 +7,4 @@ from .batching import ( # noqa: F401
|
|||||||
PretrainingBatchSamplerDataCollatorForSeq2Seq,
|
PretrainingBatchSamplerDataCollatorForSeq2Seq,
|
||||||
V2BatchSamplerDataCollatorForSeq2Seq,
|
V2BatchSamplerDataCollatorForSeq2Seq,
|
||||||
)
|
)
|
||||||
from .kd import DataCollatorForKD # noqa: F401
|
|
||||||
from .mamba import MambaDataCollator # noqa: F401
|
from .mamba import MambaDataCollator # noqa: F401
|
||||||
|
|||||||
Reference in New Issue
Block a user