Commit c8bd026e authored by VictorSanh's avatar VictorSanh
Browse files

move dependecies list to hubconf

parent 19ef2b0a
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex', 'ftfy', 'spacy']
from hubconfs.bert_hubconf import ( from hubconfs.bert_hubconf import (
bertTokenizer, bertTokenizer,
bertModel, bertModel,
......
...@@ -10,8 +10,6 @@ from pytorch_pretrained_bert.modeling import ( ...@@ -10,8 +10,6 @@ from pytorch_pretrained_bert.modeling import (
BertForTokenClassification, BertForTokenClassification,
) )
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']
# A lot of models share the same param doc. Use a decorator # A lot of models share the same param doc. Use a decorator
# to save typing # to save typing
bert_docstring = """ bert_docstring = """
......
...@@ -5,8 +5,6 @@ from pytorch_pretrained_bert.modeling_openai import ( ...@@ -5,8 +5,6 @@ from pytorch_pretrained_bert.modeling_openai import (
OpenAIGPTDoubleHeadsModel OpenAIGPTDoubleHeadsModel
) )
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex', 'ftfy', 'spacy']
# A lot of models share the same param doc. Use a decorator # A lot of models share the same param doc. Use a decorator
# to save typing # to save typing
gpt_docstring = """ gpt_docstring = """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment