diff --git a/kohya_gui/class_accelerate_launch.py b/kohya_gui/class_accelerate_launch.py index 4e63840..50d2c5c 100644 --- a/kohya_gui/class_accelerate_launch.py +++ b/kohya_gui/class_accelerate_launch.py @@ -82,17 +82,17 @@ class AccelerateLaunch: if "num_processes" in kwargs: num_processes = kwargs.get("num_processes") - if int(num_processes) > 1: + if int(num_processes) > 0: run_cmd += f" --num_processes={int(num_processes)}" if "num_machines" in kwargs: num_machines = kwargs.get("num_machines") - if int(num_machines) > 1: + if int(num_machines) > 0: run_cmd += f" --num_machines={int(num_machines)}" if "num_cpu_threads_per_process" in kwargs: num_cpu_threads_per_process = kwargs.get("num_cpu_threads_per_process") - if int(num_cpu_threads_per_process) > 1: + if int(num_cpu_threads_per_process) > 0: run_cmd += ( f" --num_cpu_threads_per_process={int(num_cpu_threads_per_process)}" )