mirror of https://github.com/bmaltais/kohya_ss
Fix configure_accelerate
parent
b27d0694c4
commit
4f190cc885
|
|
@ -119,8 +119,7 @@ def install_kohya_ss_torch2(headless: bool = False):
|
|||
"requirements_windows_torch2.txt", check_no_verify_flag=False
|
||||
)
|
||||
|
||||
if not headless:
|
||||
setup_common.configure_accelerate(run_accelerate=True)
|
||||
setup_common.configure_accelerate(run_accelerate=not headless) # False if headless is True and vice versa
|
||||
|
||||
|
||||
def install_bitsandbytes_0_35_0():
|
||||
|
|
|
|||
Loading…
Reference in New Issue