transformers/hubconf.py
2019-08-30 12:48:55 -04:00

6 lines
218 B
Python

dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']
from pytorch_transformers import (
AutoTokenizer, AutoModel, AutoModelWithLMHead, AutoModelForSequenceClassification, AutoModelForQuestionAnswering
)