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

Fix check_training_gradient_checkpointing (#34806)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-11-19 17:48:34 +01:00
committed by GitHub
parent 05ebe8b9b0
commit 469eddbe2d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_common.py
View File

@@ -847,7 +847,9 @@ class ModelTesterMixin:
] ]
or not model_class.supports_gradient_checkpointing or not model_class.supports_gradient_checkpointing
): ):
self.skipTest(reason=f"`supports_gradient_checkpointing` is False for {model_class.__name__}.") # TODO (ydshieh): use `skipTest` once pytest-dev/pytest-subtests/pull/169 is merged
# self.skipTest(reason=f"`supports_gradient_checkpointing` is False for {model_class.__name__}.")
continue
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
config.use_cache = False config.use_cache = False