Fix check_training_gradient_checkpointing (#34806)

fix

Co-authored-by: ydshieh <[email protected]>

tests/test_modeling_common.py

@@ -847,7 +847,9 @@ def check_training_gradient_checkpointing(self, gradient_checkpointing_kwargs=No
                     ]
                     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