unilm/simlm/src/collators/__init__.py

from .biencoder_collator import BiencoderCollator
from .cross_encoder_collator import CrossEncoderCollator
from .rlm_collator import DataCollatorForReplaceLM