formtgpt / data /__init__.py
Ystar124's picture
Upload DocLLM custom Transformer model
6cacf86 verified
Raw History Blame Contribute Delete
285 Bytes
"""
Data processing package for DocLLM.
"""
from data.pdf_loader import PDFLoader, PDFDocument
from data.dataset import DocPretrainDataset, DocQADataset, qa_collate_fn
__all__ = [
"PDFLoader",
"PDFDocument",
"DocPretrainDataset",
"DocQADataset",
"qa_collate_fn"
]