fix

2023-12-08 23:21:58 +08:00 · 2023-12-08 23:21:58 +08:00 · 3e9fb3d54e
parent 3febb4d3ef
commit 3e9fb3d54e
1 changed files with 1 additions and 1 deletions
--- a/mikazuki/process.py
+++ b/mikazuki/process.py
@ -21,7 +21,7 @@ def run_train(toml_path: str,
        "--config_file", toml_path,
    ]
    if multi_gpu:
-        args[3:2] = ["--multi_gpu", "--num_processes=2"]
+        args[3:3] = ["--multi_gpu", "--num_processes", "2"]

    customize_env = os.environ.copy()
    customize_env["ACCELERATE_DISABLE_RICH"] = "1"