From 3df4df868cd97587adf81eccfd12cc224e328e3f Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 6 Feb 2025 14:00:15 -0500 Subject: [PATCH] make sure to pass kwargs when using accelerate --- src/axolotl/cli/main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/axolotl/cli/main.py b/src/axolotl/cli/main.py index d1a85b929..f48b9e91e 100644 --- a/src/axolotl/cli/main.py +++ b/src/axolotl/cli/main.py @@ -200,7 +200,9 @@ def train( try: if accelerate: if cloud: - do_cli_train(cloud_config=cloud, config=config, accelerate=True) + do_cli_train( + cloud_config=cloud, config=config, accelerate=True, **kwargs + ) else: accelerate_args = [] if "main_process_port" in kwargs: