tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

make sure to pass kwargs when using accelerate

Browse Source
This commit is contained in:
Wing Lian
2025-02-06 14:00:15 -05:00
parent c82cbdc6d9
commit 3df4df868c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/cli/main.py
View File

@@ -200,7 +200,9 @@ def train(
try: try:
if accelerate: if accelerate:
if cloud: if cloud:
do_cli_train(cloud_config=cloud, config=config, accelerate=True) do_cli_train(
cloud_config=cloud, config=config, accelerate=True, **kwargs
)
else: else:
accelerate_args = [] accelerate_args = []
if "main_process_port" in kwargs: if "main_process_port" in kwargs: