From c8bd026ef6a1eb6f431d158e76cbdd8d5938ac39 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Fri, 31 May 2019 00:36:58 -0400 Subject: [PATCH] move dependecies list to hubconf --- hubconf.py | 2 ++ hubconfs/bert_hubconf.py | 2 -- hubconfs/gpt_hubconf.py | 2 -- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/hubconf.py b/hubconf.py index 0561c9a26b..2d69da8e79 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 20ae90410a..67397aeec8 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 ae8db02061..8cf64b0c02 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 = """