HubConf configuration

xijiu9 · Aug 30, 2019 · bc29aa6 · bc29aa6
1 parent f35f612
commit bc29aa6
Show file tree

Hide file tree

Showing 2 changed files with 411 additions and 3 deletions.
diff --git a/hubconf.py b/hubconf.py
@@ -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,
+)