HubConf configuration

This commit is contained in:
LysandreJik
2019-08-30 11:13:51 -04:00
parent f35f612280
commit bc29aa67a9
2 changed files with 411 additions and 3 deletions

View File

@@ -1,5 +1,10 @@
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']
from pytorch_transformers import (
AutoTokenizer, AutoModel, AutoModelWithLMHead, AutoModelForSequenceClassification, AutoModelForQuestionAnswering
)
from hubconfs.automodels_hubconf import (
autoConfig,
autoModel,
autoModelForQuestionAnswering,
autoModelForSequenceClassification,
autoModelWithLMHead,
autoTokenizer,
)