From 69e78f2a352900a2c96056ec5cccce8616fb9908 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 27 Mar 2024 16:32:41 +0100 Subject: [PATCH] update --- tests/models/qwen2/test_modeling_qwen2.py | 6 ++++++ tests/models/qwen2_moe/test_modeling_qwen2_moe.py | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/tests/models/qwen2/test_modeling_qwen2.py b/tests/models/qwen2/test_modeling_qwen2.py index 587312bfa21d..dc96f2dccf33 100644 --- a/tests/models/qwen2/test_modeling_qwen2.py +++ b/tests/models/qwen2/test_modeling_qwen2.py @@ -320,6 +320,12 @@ def is_pipeline_test_to_skip( ): return True + # Ignore copy + @require_torch_sdpa + @slow + def test_eager_matches_sdpa_generate(self): + super().test_eager_matches_sdpa_generate() + def setUp(self): self.model_tester = Qwen2ModelTester(self) self.config_tester = ConfigTester(self, config_class=Qwen2Config, hidden_size=37) diff --git a/tests/models/qwen2_moe/test_modeling_qwen2_moe.py b/tests/models/qwen2_moe/test_modeling_qwen2_moe.py index 808f8b3a8725..b6f8df27efaa 100644 --- a/tests/models/qwen2_moe/test_modeling_qwen2_moe.py +++ b/tests/models/qwen2_moe/test_modeling_qwen2_moe.py @@ -349,6 +349,12 @@ def is_pipeline_test_to_skip( ): return True + # Ignore copy + @require_torch_sdpa + @slow + def test_eager_matches_sdpa_generate(self): + super().test_eager_matches_sdpa_generate() + def setUp(self): self.model_tester = Qwen2MoeModelTester(self) self.config_tester = ConfigTester(self, config_class=Qwen2MoeConfig, hidden_size=37)