Compare commits

...

1 Commits

Author SHA1 Message Date
Wing Lian
f8bb4185bc skip s2 attention test due to timeout 2024-04-08 18:33:33 -04:00

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