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

fix

Browse Source
This commit is contained in:
Dan Saunders
2024-12-07 11:09:43 -05:00
committed by Dan Saunders
parent 541f9b39ff
commit 6f80d1d670
29 changed files with 29 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/e2e/integrations/test_cut_cross_entropy.py
View File

@@ -4,7 +4,7 @@ Simple end-to-end test for Cut Cross Entropy integration
import pytest
from axolotl.cli import load_datasets
from axolotl.cli.datasets import load_datasets
from axolotl.common.cli import TrainerCliArgs
from axolotl.train import train
from axolotl.utils import get_pytorch_version

2
tests/e2e/integrations/test_liger.py
View File

@@ -4,7 +4,7 @@ Simple end-to-end test for Liger integration
from e2e.utils import require_torch_2_4_1
from axolotl.cli import load_datasets
from axolotl.cli.datasets import load_datasets
from axolotl.common.cli import TrainerCliArgs
from axolotl.train import train
from axolotl.utils.config import normalize_config, prepare_plugins