diff --git a/hubconf.py b/hubconf.py index 0561c9a26b2..2d69da8e796 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,3 +1,5 @@ +dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex', 'ftfy', 'spacy'] + from hubconfs.bert_hubconf import ( bertTokenizer, bertModel, diff --git a/hubconfs/bert_hubconf.py b/hubconfs/bert_hubconf.py index 20ae90410a8..67397aeec8b 100644 --- a/hubconfs/bert_hubconf.py +++ b/hubconfs/bert_hubconf.py @@ -10,8 +10,6 @@ from pytorch_pretrained_bert.modeling import ( BertForTokenClassification, ) -dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex'] - # A lot of models share the same param doc. Use a decorator # to save typing bert_docstring = """ diff --git a/hubconfs/gpt_hubconf.py b/hubconfs/gpt_hubconf.py index ae8db020615..8cf64b0c02e 100644 --- a/hubconfs/gpt_hubconf.py +++ b/hubconfs/gpt_hubconf.py @@ -5,8 +5,6 @@ from pytorch_pretrained_bert.modeling_openai import ( OpenAIGPTDoubleHeadsModel ) -dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex', 'ftfy', 'spacy'] - # A lot of models share the same param doc. Use a decorator # to save typing gpt_docstring = """