From ae494776e45c92897be68b813d2e7b1a00649866 Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Tue, 17 Dec 2024 13:52:34 +0000 Subject: [PATCH] pre-commit fix --- src/axolotl/cli/main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/axolotl/cli/main.py b/src/axolotl/cli/main.py index 87f0f7a0c..f3032a9d6 100644 --- a/src/axolotl/cli/main.py +++ b/src/axolotl/cli/main.py @@ -12,7 +12,12 @@ from axolotl.cli.utils import ( build_command, fetch_from_github, ) -from axolotl.common.cli import ConvertDiffTransformerCliArgs, EvaluateCliArgs, PreprocessCliArgs, TrainerCliArgs +from axolotl.common.cli import ( + ConvertDiffTransformerCliArgs, + EvaluateCliArgs, + PreprocessCliArgs, + TrainerCliArgs, +) from axolotl.utils import set_pytorch_cuda_alloc_conf from axolotl.utils.config.models.input.v0_4_1 import AxolotlInputConfig