From f8bb4185bce62ff7b744919fca032fdc0f36a327 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 8 Apr 2024 18:33:33 -0400 Subject: [PATCH] skip s2 attention test due to timeout --- tests/e2e/patched/test_llama_s2_attention.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/e2e/patched/test_llama_s2_attention.py b/tests/e2e/patched/test_llama_s2_attention.py index f1d37eb3c..010df9f99 100644 --- a/tests/e2e/patched/test_llama_s2_attention.py +++ b/tests/e2e/patched/test_llama_s2_attention.py @@ -7,6 +7,8 @@ import os import unittest from pathlib import Path +import pytest + from axolotl.cli import load_datasets from axolotl.common.cli import TrainerCliArgs from axolotl.train import train @@ -19,6 +21,7 @@ LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" +@pytest.mark.skip("Skipping test due to timeout.") class TestLlamaShiftedSparseAttention(unittest.TestCase): """ Test case for Llama models using S2 Attn