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

update loading logics

Browse Source
This commit is contained in:
thomwolf
2019-09-24 14:03:58 +02:00
parent 271f213621
commit f5397ffc3b
4 changed files with 55 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch_transformers/file_utils.py
View File

@@ -49,6 +49,7 @@ except (AttributeError, ImportError):
PYTORCH_TRANSFORMERS_CACHE = PYTORCH_PRETRAINED_BERT_CACHE # Kept for backward compatibility
WEIGHTS_NAME = "pytorch_model.bin"
TF2_WEIGHTS_NAME = 'tf_model.h5'
TF_WEIGHTS_NAME = 'model.ckpt'
CONFIG_NAME = "config.json"