Making the configurations classes so it is clear what is overwritten …

…at runtime.
JoePenna · Feb 28, 2023 · 0a217fc · 0a217fc
1 parent 0c97578
commit 0a217fc
Show file tree

Hide file tree

Showing 21 changed files with 925 additions and 1,784 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,8 +1,13 @@
-
 .DS_Store
 gen.bat
 gen_ref.bat
 train.bat
 __pycache__
 */**/__pycache__
-logs
+logs
+venv
+.idea
+regularization_images
+training_images
+*.egg-info
+*.ckpt
diff --git a/README.md b/README.md
@@ -3,7 +3,9 @@ I am testing breaking changes in this repo.  Fork/clone at your own risk.
 
 - `python -m venv venv`
 - `venv\Scripts\activate.bat`
-- `...do stuff...`
+- `pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117`
+- `pip install -r requirements.txt`
+- `python "main.py" --base "C:\Users\David\Dev\david-bielejeski-Dreambooth-Stable-Diffusion\configs\stable-diffusion\v1-finetune_unfrozen.yaml" -t --actual_resume "C:\Users\David\Dev\david-bielejeski-Dreambooth-Stable-Diffusion\v1-5-pruned-emaonly-pruned.ckpt" --reg_data_root "C:\Users\David\Dev\david-bielejeski-Dreambooth-Stable-Diffusion\regularization_images" -n "test" --gpus 0, --data_root "C:\Users\David\Dev\david-bielejeski-Dreambooth-Stable-Diffusion\training_images" --max_training_steps 100 --class_word "david" --token "davidbielejeski" --no-test --flip_p 0`
 - `deactivate`
 
 # Index

diff --git a/configs/stable-diffusion/v1-finetune.yaml b/configs/stable-diffusion/v1-finetune.yaml
diff --git a/configs/stable-diffusion/v1-finetune_unfrozen.yaml b/configs/stable-diffusion/v1-finetune_unfrozen.yaml
diff --git a/configs/stable-diffusion/v1-finetune_unfrozen_save_checkpoints_every_250_steps.yaml b/configs/stable-diffusion/v1-finetune_unfrozen_save_checkpoints_every_250_steps.yaml