Fix

2025-01-10 17:32:33 +00:00
parent 210c58a4db
commit c9e37496cb
26 changed files with 26 additions and 26 deletions
--- a/tests/e2e/test_mixtral.py
+++ b/tests/e2e/test_mixtral.py
@@ -9,7 +9,7 @@ import unittest
 import torch
 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.train import train
 from axolotl.utils.config import normalize_config