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

[WIP] Ensure TF model configs can be converted to proper JSON (#14415)

Browse Source 
* test: make sure model configs are jsonifiable

* fix: return python dict instead of config object

* fix: accept pretrained config and use correct class

* Re-enabling slow tests and applying them to core models only

* Re-enabling slow tests and applying them to core models only

* Add new test file to fetcher

* Remove tooslow tests from test_modeling_tf_common.py

* make style

* Style fixes

* Style fixes

* Style fixes

* Style fixes

* Adding core tests to GPT2 and BART

* Removing unused imports

Co-authored-by: niklas.fruehauf <niklas.fruehauf@sovanta.com>
Co-authored-by: matt <rocketknight1@gmail.com>
This commit is contained in:
N
2021-11-17 21:24:39 +01:00
committed by GitHub
parent 754202de4f
commit 1991da07f7
7 changed files with 374 additions and 223 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/tests_fetcher.py
View File

@@ -252,7 +252,7 @@ SPECIAL_MODULE_TO_TEST_MAP = {
"file_utils.py": ["test_file_utils.py", "test_model_output.py"],
"modelcard.py": "test_model_card.py",
"modeling_flax_utils.py": "test_modeling_flax_common.py",
"modeling_tf_utils.py": "test_modeling_tf_common.py",
"modeling_tf_utils.py": ["test_modeling_tf_common.py", "test_modeling_tf_core.py"],
"modeling_utils.py": ["test_modeling_common.py", "test_offline.py"],
"models/auto/modeling_auto.py": ["test_modeling_auto.py", "test_modeling_tf_pytorch.py", "test_modeling_bort.py"],
"models/auto/modeling_flax_auto.py": "test_flax_auto.py",