hubconf.py 218 Bytes
Newer Older
VictorSanh's avatar
VictorSanh committed
1
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']
VictorSanh's avatar
VictorSanh committed
2

LysandreJik's avatar
LysandreJik committed
3
4
from pytorch_transformers import (
    AutoTokenizer, AutoModel, AutoModelWithLMHead, AutoModelForSequenceClassification, AutoModelForQuestionAnswering
VictorSanh's avatar
VictorSanh committed
5
)