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
2025-01-10 17:32:33 +00:00
parent 210c58a4db
commit c9e37496cb
26 changed files with 26 additions and 26 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 import pytest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils import get_pytorch_version from axolotl.utils import get_pytorch_version

2
tests/e2e/patched/test_4d_multipack_llama.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_fa_xentropy.py
View File

@@ -8,7 +8,7 @@ import os
import pytest import pytest
from transformers.utils import is_torch_bf16_gpu_available from transformers.utils import is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_falcon_samplepack.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_fused_llama.py
View File

@@ -9,7 +9,7 @@ import unittest
import pytest import pytest
from transformers.utils import is_torch_bf16_gpu_available from transformers.utils import is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

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

@@ -8,7 +8,7 @@ import unittest
import pytest import pytest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_lora_llama_multipack.py
View File

@@ -9,7 +9,7 @@ import unittest
import pytest import pytest
from transformers.utils import is_auto_gptq_available, is_torch_bf16_gpu_available from transformers.utils import is_auto_gptq_available, is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_mistral_samplepack.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_mixtral_samplepack.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_phi_multipack.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_resume.py
View File

@@ -9,7 +9,7 @@ import subprocess
from transformers.utils import is_torch_bf16_gpu_available from transformers.utils import is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/patched/test_unsloth_qlora.py
View File

@@ -6,7 +6,7 @@ import os
import pytest import pytest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_embeddings_lr.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_falcon.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_llama.py
View File

@@ -7,7 +7,7 @@ import os
from e2e.utils import check_model_output_exists from e2e.utils import check_model_output_exists
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_llama_pretrain.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_llama_vision.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_lora_llama.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_mamba.py
View File

@@ -8,7 +8,7 @@ import unittest
import pytest import pytest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_mistral.py
View File

@@ -8,7 +8,7 @@ import unittest
from transformers.utils import is_torch_bf16_gpu_available from transformers.utils import is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_mixtral.py
View File

@@ -9,7 +9,7 @@ import unittest
import torch import torch
from transformers.utils import is_torch_bf16_gpu_available from transformers.utils import is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_optimizers.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_packing_loss.py
View File

@@ -8,7 +8,7 @@ import unittest
from transformers.utils import is_torch_bf16_gpu_available from transformers.utils import is_torch_bf16_gpu_available
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_phi.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_relora_llama.py
View File

@@ -7,7 +7,7 @@ import os
import unittest import unittest
from pathlib import Path from pathlib import Path
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config

2
tests/e2e/test_reward_model_llama.py
View File

@@ -6,7 +6,7 @@ import logging
import os import os
import unittest import unittest
from axolotl.common.cli import TrainerCliArgs from axolotl.cli.args import TrainerCliArgs
from axolotl.common.datasets import load_datasets from axolotl.common.datasets import load_datasets
from axolotl.train import train from axolotl.train import train
from axolotl.utils.config import normalize_config from axolotl.utils.config import normalize_config