From 59b1fa2381b5c119702d248cfbbd9db1e0f9eb22 Mon Sep 17 00:00:00 2001 From: JB Lau <1557853+hackyon@users.noreply.github.com> Date: Wed, 14 Feb 2024 12:44:15 -0500 Subject: [PATCH] testing inference --- tests/test_modeling_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index c969c7c0d33b08..02a04ea0ebab9c 100755 --- 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")