mi250 new skip from update

ROCm · Jul 29, 2024 · da06f2a · da06f2a
1 parent 106a969
commit da06f2a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/models/depth_anything/test_modeling_depth_anything.py b/tests/models/depth_anything/test_modeling_depth_anything.py
@@ -18,7 +18,7 @@
 
 from transformers import DepthAnythingConfig, Dinov2Config
 from transformers.file_utils import is_torch_available, is_vision_available
-from transformers.testing_utils import require_torch, require_vision, slow, torch_device
+from transformers.testing_utils import require_torch, require_vision, slow, torch_device, skipIfRocm
 
 from ...test_configuration_common import ConfigTester
 from ...test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor
@@ -156,6 +156,10 @@ def setUp(self):
             common_properties=["patch_size"],
         )
 
+    @skipIfRocm
+    def test_batching_equivalence(self):
+        super().test_batching_equivalence()
+
     def test_config(self):
         self.config_tester.run_common_tests()