tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

skip s2 attention test due to timeout

Browse Source
This commit is contained in:
Wing Lian
2024-04-08 18:33:33 -04:00
parent 2fa65b9599
commit f8bb4185bc
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/e2e/patched/test_llama_s2_attention.py
View File

@@ -7,6 +7,8 @@ import os
import unittest import unittest
from pathlib import Path from pathlib import Path
import pytest
from axolotl.cli import load_datasets from axolotl.cli import load_datasets
from axolotl.common.cli import TrainerCliArgs from axolotl.common.cli import TrainerCliArgs
from axolotl.train import train from axolotl.train import train
@@ -19,6 +21,7 @@ LOG = logging.getLogger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true" os.environ["WANDB_DISABLED"] = "true"
@pytest.mark.skip("Skipping test due to timeout.")
class TestLlamaShiftedSparseAttention(unittest.TestCase): class TestLlamaShiftedSparseAttention(unittest.TestCase):
""" """
Test case for Llama models using S2 Attn Test case for Llama models using S2 Attn