File size: 166 Bytes
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