testing inference

huggingface · Feb 14, 2024 · 59b1fa2 · 59b1fa2
1 parent e454391
commit 59b1fa2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -3444,7 +3444,7 @@ def test_flash_attn_2_generate_padding_right(self):
 
     @parameterized.expand([("float16",), ("bfloat16",), ("float32",)])
     @require_torch_sdpa
-    @slow
+    #@slow
     def test_eager_matches_sdpa_inference(self, torch_dtype: str):
         if not self.all_model_classes[0]._supports_sdpa:
             self.skipTest(f"{self.all_model_classes[0].__name__} does not support SDPA")