Thesis/pretrain/models/__init__.py

7 lines
124 B
Python
Executable File

from transformers import BartForConditionalGeneration, T5ForConditionalGeneration, GPT2LMHeadModel
from .model import *