data_collator_for_seq_to_seq module