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

pinning transformers version

Browse Source
This commit is contained in:
Salman Mohammadi
2025-04-08 17:17:23 +01:00
parent b98dbafc31
commit e1a8dfbe8c
7 changed files with 111 additions and 208 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/e2e/solo/test_flex.py
View File

@@ -14,7 +14,7 @@ from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from ..utils import check_tensorboard, require_torch_2_6_0, with_temp_dir
from ..utils import check_tensorboard, require_torch_2_7_0, with_temp_dir
LOG = logging.getLogger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
@@ -25,7 +25,7 @@ class TestPackedFlex(unittest.TestCase):
Test case for Packed training of llama models
"""
@require_torch_2_6_0
@require_torch_2_7_0
@with_temp_dir
def test_loss_llama(self, temp_dir):
# pylint: disable=duplicate-code