c8ddb9b
1
2
3
4
5
6
"""Dataset and custom collate function to load""" from .collate import custom_collate from .datasets import TextImageDataset from .tokenizer import TAIMGANTokenizer