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

Fix Torch.hub + Integration test

Browse Source
This commit is contained in:
Julien Chaumond
2020-04-21 10:49:00 -04:00
parent 7d40901ce3
commit d32585a304
6 changed files with 43 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
hubconf.py
View File

@@ -1,4 +1,4 @@
from transformers import (
from src.transformers import (
AutoConfig,
AutoModel,
AutoModelForQuestionAnswering,
@@ -6,10 +6,10 @@ from transformers import (
AutoModelWithLMHead,
AutoTokenizer,
)
from transformers.file_utils import add_start_docstrings
from src.transformers.file_utils import add_start_docstrings
dependencies = ["torch", "tqdm", "boto3", "requests", "regex", "sentencepiece", "sacremoses"]
dependencies = ["torch", "numpy", "tokenizers", "boto3", "filelock", "requests", "tqdm", "regex", "sentencepiece", "sacremoses"]
@add_start_docstrings(AutoConfig.__doc__)