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

precommit

Browse Source
This commit is contained in:
Dan Saunders
2025-03-21 16:43:14 +00:00
parent ce35b2a95f
commit 4ac65462f0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/e2e/patched/test_sp.py
View File

@@ -175,7 +175,7 @@ def test_sequence_parallel_slicing(
def test_config_validation_with_valid_inputs(cfg):
"""Test that valid sequence parallelism configurations pass validation."""
# Import the actual model class with appropriate mocks
from axolotl.utils.config.models.input.v0_4_1 import AxolotlInputConfig
from axolotl.utils.schemas.config import AxolotlInputConfig
# Valid configuration: sequence_parallel_degree > 1 and flash_attention is True
cfg = cfg | {
@@ -191,7 +191,7 @@ def test_config_validation_with_valid_inputs(cfg):
def test_config_validation_with_invalid_inputs(cfg):
"""Test that invalid sequence parallelism configurations fail validation."""
from axolotl.utils.config.models.input.v0_4_1 import AxolotlInputConfig
from axolotl.utils.schemas.config import AxolotlInputConfig
# Invalid configuration: sequence_parallel_degree > 1 but flash_attention is False
cfg = cfg | {