saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

style

Browse source
This commit is contained in:
Cyril Vallez 2025-02-07 11:41:07 +01:00
parent c95dc4ebe7
commit c341f8c6fb
No known key found for this signature in database
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_modeling_common.py
View file

@ -2360,7 +2360,7 @@ class ModelTesterMixin:
else:
expected_missing = set()
for pattern in model_reloaded._keys_to_ignore_on_load_missing:
expected_missing.update({k for k in model_reloaded.state_dict().keys() if re.search(pattern, k) is not None})
expected_missing.update({k for k in param_names if re.search(pattern, k) is not None})
self.assertEqual(
missed_missing,
expected_missing,