Commit Graph

351 Commits

Author SHA1 Message Date
Augusto de la Torre 0716c40ab6 Add support for enhanced dataset configuration
Add support for:
* flip_p.txt
* cond_dropout.txt
* local.yaml consolidated config (including default captions)
* global.yaml consolidated config which applies recursively to subfolders
* flip_p, and cond_dropout config per image
* manifest.json with full image-level configuration
2023-03-08 15:02:14 +01:00
Victor Hall 2e3d044ba3 Merge branch 'main' of https://github.com/victorchall/EveryDream2trainer 2023-03-04 15:09:31 -05:00
Victor Hall 87ffb5413e log warning when val/loss is showing upward trend 2023-03-04 15:08:47 -05:00
Victor Hall 32585a74b2
Merge pull request #103 from damian0815/feat_text_encoder_LR
Separate text encoder LR; logging & pbar improvements; misc sample generator tweaks
2023-03-03 17:46:45 -05:00
Damian Stewart 29c93eca03 if progress bars are disabled, log a short message instead 2023-03-03 10:50:48 +01:00
Damian Stewart 644c9e6b2a log to logging.info instead of stdout 2023-03-03 09:52:44 +01:00
Damian Stewart c4d37862ba document new sample generator params 2023-03-02 23:12:47 +01:00
Damian Stewart 15187ae2e2 fix log leaking color 2023-03-02 22:53:59 +01:00
Damian Stewart e2fd45737d overwrite args.seed with the actual seed if -1 is passed (so it appears in tensorboard)
also improve logging when unet training is disabled
2023-03-02 22:52:33 +01:00
Damian Stewart bb1fbb329a tweak docs 2023-03-02 22:45:22 +01:00
Damian Stewart 97a8a49773 don't log separate text encoder LR if it's the same as unet LR 2023-03-02 22:36:32 +01:00
Damian Stewart ae281976ca documentation of new text encoder LR and aspect_ratio settings 2023-03-02 22:36:11 +01:00
Damian Stewart fe0083877f add aspect_ratio arg to sample generation 2023-03-02 22:16:21 +01:00
Damian Stewart 52c722714c Merge branch 'feat_text_encoder_LR' of github.com:damian0815/EveryDream2trainer into feat_text_encoder_LR 2023-03-02 20:02:38 +01:00
damian 61558be2ae logging and progress bar improvements 2023-03-02 18:29:28 +01:00
Damian Stewart 17c61db5ca wip aspect ratio samples 2023-03-02 14:50:17 +01:00
Damian Stewart 8100e42159 fix issues and improve sample generator 2023-03-02 13:03:50 +01:00
Damian Stewart c82664b3f3 add text encoder LR setting to optimizer.json 2023-03-02 00:13:43 +01:00
Victor Hall ba87b0cae1 log how lr was derived 2023-03-01 12:43:32 -05:00
Victor Hall cb7f3d26aa start troubleshooting doc 2023-03-01 12:40:36 -05:00
Victor Hall 8ed8c20485 Merge branch 'main' of https://github.com/victorchall/EveryDream2trainer 2023-03-01 12:26:50 -05:00
Victor Hall 600eaa404d make sure main lr arg overrides optimizer.json 2023-03-01 12:26:36 -05:00
Victor Hall b0f7777a6c
Merge pull request #98 from nawnie/main
revised pr
2023-02-28 23:53:35 -05:00
Victor Hall b5fad8c675 docs 2023-02-28 21:19:52 -05:00
Victor Hall a5f023805d Merge branch 'main' of https://github.com/victorchall/EveryDream2trainer 2023-02-28 21:19:22 -05:00
Victor Hall 8abef6bc74 revert multiline txt for now due to bug 2023-02-28 21:14:19 -05:00
Victor Hall c446194599 sys req on main readme 2023-02-28 10:57:49 -05:00
nawnie 893eb15cf8 Created using Colaboratory 2023-02-28 09:35:12 -06:00
nawnie 97450d8881
Update Train_Colab.ipynb 2023-02-28 09:33:29 -06:00
nawnie db4e632b3f
Update Train_Colab.ipynb 2023-02-28 09:30:03 -06:00
nawnie 0f3fde5899 Created using Colaboratory 2023-02-28 09:27:51 -06:00
nawnie 308c42ebd0
default fix
left adv samples and disable text on
2023-02-28 08:47:25 -06:00
nawnie d12e39a621
Update sample_prompts.json 2023-02-27 19:57:08 -06:00
nawnie d40d980bce Created using Colaboratory 2023-02-27 19:08:50 -06:00
nawnie 9961522001
sample_prompts.json
add comments to explain how to better modify your samples
2023-02-27 18:22:58 -06:00
nawnie e9ba16f544
sample updates
add more samplers
2023-02-27 18:14:23 -06:00
Victor Hall 11cea762c4
Merge pull request #96 from qslug/docker
Streamline Vast/Runpod docker
2023-02-27 13:25:24 -05:00
Augusto de la Torre deb83f42cb Streamline Vast/Runpod docker
This removes runtime installation steps to make vast.ai work out of the box and fixes some issues with dev containers
2023-02-27 02:50:57 +01:00
Victor Hall 8c6d6bccfe finish notdreambooth.md 2023-02-26 19:24:22 -05:00
Victor Hall 952e34a4a8 merge optimizer.json 2023-02-26 19:12:42 -05:00
Victor Hall 6c87af7711 doc, fix lr defaulting bug, notdreambooth.md, read multicaption from .txt by line 2023-02-26 19:11:42 -05:00
Victor Hall 329ba2bfc1
fix bug with reading lr from optimizer.json 2023-02-26 10:00:42 -05:00
Victor Hall 2acc261818 add lion install to colab notebook 2023-02-25 16:46:43 -05:00
Victor Hall 0b037348ad update coco example cfg 2023-02-25 16:37:10 -05:00
Victor Hall d049dc684f add lion to windows install 2023-02-25 16:36:04 -05:00
Victor Hall f4999281ac
Merge pull request #93 from victorchall/lion
log lr with rest of optimizer settings to console, put lr into optimi…
2023-02-25 16:24:27 -05:00
Victor Hall b7de987f4c log lr with rest of optimizer settings to console, put lr into optimizer.json default 2023-02-25 16:23:33 -05:00
Victor Hall e7fc71ffa1
Merge pull request #92 from victorchall/lion
fix mem leak on huge data, rework optimizer to separate json, add lio…
2023-02-25 16:20:10 -05:00
Victor Hall 1f80cb4b68 optimizer doc 2023-02-25 16:19:23 -05:00
Victor Hall a9b0189947 fix mem leak on huge data, rework optimizer to separate json, add lion optimizer 2023-02-25 15:05:22 -05:00