Thesis/pretrain/models/__init__.py

7 lines
124 B
Python
Raw Normal View History

2022-12-26 04:54:46 +00:00
from transformers import BartForConditionalGeneration, T5ForConditionalGeneration, GPT2LMHeadModel
from .model import *