diff --git a/.gitignore b/.gitignore index 91e52ac..5985023 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ venv +venv1 mytraining.ps __pycache__ diff --git a/README.md b/README.md index 41594f3..7deeebd 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,7 @@ # HOWTO This repo provide all the required config to run the Dreambooth version found in this note: https://note.com/kohya_ss/n/nee3ed1649fb6 - +The setup of bitsandbytes with Adam8bit support for windows: https://note.com/kohya_ss/n/n47f654dc161e ## Required Dependencies @@ -23,25 +23,25 @@ Open a regular Powershell terminal and type the following inside: ```powershell git clone https://github.com/bmaltais/kohya_ss.git cd kohya_ss + 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 diffusers -pip install -r requirements.txt -pip install OmegaConf -pip install pytorch_lightning - +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 -# Setup bitsandbytes with Adam8bit support for windows: https://note.com/kohya_ss/n/n47f654dc161e -pip install bitsandbytes==0.35.0 cp .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\ cp .\bitsandbytes_windows\cextension.py .\venv\Lib\site-packages\bitsandbytes\cextension.py cp .\bitsandbytes_windows\main.py .\venv\Lib\site-packages\bitsandbytes\cuda_setup\main.py -accelerate config: +accelerate config + +``` + +Answers to accelerate config: + +```txt - 0 - 0 - NO diff --git a/requirements.txt b/requirements.txt index f1fed2e..4c69950 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,7 @@ ftfy albumentations opencv-python einops -diffusers[torch]==0.9.0 \ No newline at end of file +diffusers[torch]==0.9.0 +pytorch_lightning +bitsandbytes==0.35.0 +tensorboard \ No newline at end of file