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

[hubconf] Modify pythonpath to get canonical imports to work

Browse Source 
See https://github.com/huggingface/transformers/pull/3881/files#r412292660

Should we remove SRC_DIR from sys.path right after the imports, @aaugustin?
This commit is contained in:
Julien Chaumond
2020-04-23 16:27:43 -04:00
parent 77b75d2c78
commit cbbb3c43c5
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
hubconf.py
View File

@@ -1,12 +1,19 @@
from src.transformers import ( import os
import sys
SRC_DIR = os.path.join(os.path.dirname(__file__), "src")
sys.path.append(SRC_DIR)
from transformers import (
AutoConfig, AutoConfig,
AutoModel, AutoModel,
AutoModelForQuestionAnswering, AutoModelForQuestionAnswering,
AutoModelForSequenceClassification, AutoModelForSequenceClassification,
AutoModelWithLMHead, AutoModelWithLMHead,
AutoTokenizer, AutoTokenizer,
add_start_docstrings,
) )
from src.transformers.file_utils import add_start_docstrings
dependencies = ["torch", "numpy", "tokenizers", "boto3", "filelock", "requests", "tqdm", "regex", "sentencepiece", "sacremoses"] dependencies = ["torch", "numpy", "tokenizers", "boto3", "filelock", "requests", "tqdm", "regex", "sentencepiece", "sacremoses"]