update names of torch.hub to simpler names - update docstring

2019-08-30 22:20:44 +02:00
parent c665e0fcfe
commit 7a1f174a9d
2 changed files with 61 additions and 358 deletions
--- a/hubconf.py
+++ b/hubconf.py
@@ -1,10 +1,10 @@
-dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']
+dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex', 'sentencepiece', 'sacremoses']

 from hubconfs.automodels_hubconf import (
-    autoConfig,
-    autoModel,
-    autoModelForQuestionAnswering,
-    autoModelForSequenceClassification,
-    autoModelWithLMHead,
-    autoTokenizer,
+    config,
+    model,
+    modelForQuestionAnswering,
+    modelForSequenceClassification,
+    modelWithLMHead,
+    tokenizer,
 )