0ab2566
1
2
3
4
5
6
7
8
9
10
11
12
COLLATORS = {} def register_collator(name): def register_collator_cls(cls): if name in COLLATORS: return COLLATORS[name] COLLATORS[name] = cls return cls return register_collator_cls from .qwen_vl_rerank import TrainRerankDataCollator
