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

Fix _configuration_file argument getting passed to model (#15629)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-02-11 13:46:08 -05:00
committed by GitHub
parent 85aee09e9a
commit 2dce350b33
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_configuration_common.py
View File

@@ -334,8 +334,12 @@ class ConfigurationVersioningTest(unittest.TestCase):
import transformers as new_transformers
new_transformers.configuration_utils.__version__ = "v4.0.0"
new_configuration = new_transformers.models.auto.AutoConfig.from_pretrained(repo)
new_configuration, kwargs = new_transformers.models.auto.AutoConfig.from_pretrained(
repo, return_unused_kwargs=True
)
self.assertEqual(new_configuration.hidden_size, 2)
# This checks `_configuration_file` ia not kept in the kwargs by mistake.
self.assertDictEqual(kwargs, {"_from_auto": True})
# Testing an older version by monkey-patching the version in the module it's used.
import transformers as old_transformers