Merge pull request #5 from kohya-ss/dev

Merge dev to main.

README-ja.md

@@ -53,7 +53,7 @@ python -m venv --system-site-packages venv
 .\venv\Scripts\activate
 
 pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
-pip install --upgrade -r requirements_db_finetune.txt
+pip install --upgrade -r requirements.txt
 pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
 
 cp .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\
@@ -74,7 +74,7 @@ python -m venv --system-site-packages venv
 .\venv\Scripts\activate
 
 pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
-pip install --upgrade -r requirements_db_finetune.txt
+pip install --upgrade -r requirements.txt
 pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
 
 copy /y .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\

README.md

@@ -52,7 +52,7 @@ python -m venv --system-site-packages venv
 .\venv\Scripts\activate
 
 pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
-pip install --upgrade -r requirements_db_finetune.txt
+pip install --upgrade -r requirements.txt
 pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
 
 cp .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\
@@ -82,7 +82,7 @@ When a new release comes out you can upgrade your repo with the following comman
 cd sd-scripts
 git pull
 .\venv\Scripts\activate
-pip install --upgrade -r <requirement file name>
+pip install --upgrade -r requirements.txt
 ```
 
 Once the commands have completed successfully you should be ready to use the new version.

requirements.txt

@@ -0,0 +1,23 @@
+accelerate==0.15.0
+transformers==4.25.1
+ftfy
+albumentations
+opencv-python
+einops
+diffusers[torch]==0.10.2
+pytorch_lightning
+bitsandbytes==0.35.0
+tensorboard
+safetensors==0.2.6
+gradio
+altair
+easygui
+# for BLIP captioning
+requests
+timm==0.4.12
+fairscale==0.4.4
+# for WD14 captioning
+tensorflow<2.11
+huggingface-hub
+# for kohya_ss library
+.

requirements_blip.txt

@@ -1,4 +0,0 @@
-timm==0.4.12
-transformers==4.16.2
-fairscale==0.4.4
-.

requirements_db_finetune.txt

@@ -1,12 +0,0 @@
-accelerate==0.14.0
-transformers>=4.21.0
-ftfy
-albumentations
-opencv-python
-einops
-pytorch_lightning
-safetensors
-bitsandbytes==0.35.0
-tensorboard
-diffusers[torch]==0.10.2
-.

requirements_wd14_tagger.txt

@@ -1,3 +0,0 @@
-tensorflow<2.11
-huggingface-hub
-.