1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revving back bitsandbytes for linux causing training issues

pull/1080/head
bmaltais 2023-06-26 15:13:51 -04:00
parent 1db40bba63
commit caa22aeb6d
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements_linux.txt
View File

@ -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.39.1
xformers==0.0.20 bitsandbytes==0.35.0
accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0
-r requirements.txt

2
requirements_linux_docker.txt
View File

@ -1,5 +1,5 @@
xformers==0.0.20
bitsandbytes==0.39.1
bitsandbytes==0.35.0
accelerate==0.15.0
tensorboard==2.12.1
tensorflow==2.12.0

4
requirements_runpod.txt
View File

@ -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.39.1
--no-cache-dir 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
accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0
-r requirements.txt