make sure to pass kwargs when using accelerate

2025-02-06 14:00:15 -05:00
parent c82cbdc6d9
commit 3df4df868c
1 changed files with 3 additions and 1 deletions
--- 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: