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: