HubConf configuration

bc29aa67 · LysandreJik · f35f6122 · bc29aa67 · bc29aa67
Commit bc29aa67 authored Aug 30, 2019 by LysandreJik
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 411 additions and 3 deletions

hubconf.py hubconf.py +8 -3

hubconfs/automodels_hubconf.py hubconfs/automodels_hubconf.py +403 -0

No files found.
--- a/hubconf.py
+++ b/hubconf.py
 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,
+)
\ No newline at end of file
--- a/hubconfs/automodels_hubconf.py
+++ b/hubconfs/automodels_hubconf.py