SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

move dependecies list to hubconf

Browse Source
This commit is contained in:
VictorSanh
2019-05-31 00:36:58 -04:00
parent 19ef2b0a66
commit c8bd026ef6
3 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
hubconf.py
View File

@@ -1,3 +1,5 @@
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex', 'ftfy', 'spacy']
from hubconfs.bert_hubconf import ( from hubconfs.bert_hubconf import (
bertTokenizer, bertTokenizer,
bertModel, bertModel,

2
hubconfs/bert_hubconf.py
View File

@@ -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 = """

2
hubconfs/gpt_hubconf.py
View File

@@ -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 = """