From 469eddbe2d7d052d7a7b8ebcdc619f17dbb55a47 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 19 Nov 2024 17:48:34 +0100 Subject: [PATCH] Fix `check_training_gradient_checkpointing` (#34806) fix Co-authored-by: ydshieh --- tests/test_modeling_common.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 3ef30fc8ae..4cfc91aade 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -847,7 +847,9 @@ class ModelTesterMixin: ] 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.use_cache = False