riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-24 23:08:57 +06:00
Code Issues Actions 35 Packages Projects Releases Wiki Activity

[Flax] Fix flax test save pretrained (#12256)

Browse Source 
* fix_torch_device_generate_test

* remove @

* fix flax save pretrained test
This commit is contained in:
Patrick von Platen 2021-06-21 16:37:13 +01:00 committed by GitHub
parent b75b5605c9
commit 4e9a6796c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_flax_common.py
View File

@ -247,9 +247,6 @@ class FlaxModelTesterMixin:
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
if model_class.__name__ != "FlaxBertModel":
continue
with self.subTest(model_class.__name__): with self.subTest(model_class.__name__):
model = model_class(config) model = model_class(config)