Updated to bitsandbytes version for linux and windows

requirements_linux.txt

@@ -1,4 +1,4 @@
 torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
-xformers==0.0.20 bitsandbytes==0.35.0
+xformers==0.0.20 bitsandbytes==0.41.0
 tensorboard==2.12.3 tensorflow==2.12.0
 -r requirements.txt

requirements_runpod.txt

@@ -1,5 +1,5 @@
 torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
-xformers==0.0.20 bitsandbytes==0.35.0
+xformers==0.0.20 bitsandbytes==0.41.0
 tensorboard==2.12.3 tensorflow==2.12.0 wheel
 tensorrt
 -r requirements.txt

requirements_windows_torch2.txt

@@ -1,5 +1,5 @@
 torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118 # no_verify
 xformers==0.0.20
-https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.39.1-py3-none-win_amd64.whl # no_verify
+https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.40.1.post1-py3-none-win_amd64.whl # no_verify
 tensorboard==2.12.3 tensorflow==2.12.0
 -r requirements.txt

setup/setup_windows.py

@@ -144,7 +144,7 @@ def install_kohya_ss_torch2():
     # )
     setup_common.install_requirements('requirements_windows_torch2.txt', check_no_verify_flag=False)
     # install('https://huggingface.co/r4ziel/xformers_pre_built/resolve/main/triton-2.0.0-cp310-cp310-win_amd64.whl', 'triton', reinstall=reinstall)
-    sync_bits_and_bytes_files()
+    # sync_bits_and_bytes_files()
     setup_common.configure_accelerate(run_accelerate=True)
     # run_cmd(f'accelerate config')