From 4c050ce807f033e903f3d3af45954eae8ef4b721 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 c37aa5484..e922d37e9 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