6c87af7711
doc, fix lr defaulting bug, notdreambooth.md, read multicaption from .txt by line
2023-02-26 19:11:42 -05:00
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
b7de987f4c
log lr with rest of optimizer settings to console, put lr into optimizer.json default
2023-02-25 16:23:33 -05:00
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
a9b0189947
fix mem leak on huge data, rework optimizer to separate json, add lion optimizer
2023-02-25 15:05:22 -05:00
4646af7434
use StableDiffusionPipeline.from_pretrained() to download the model
2023-02-20 22:00:25 +01:00
60702ee040
some validation tweaks and fixes
2023-02-19 10:20:16 +01:00
9365d5a7b2
restore saving on error
2023-02-19 00:21:20 +01:00
80c6883729
default to 2perc zero freq noise
2023-02-18 15:17:08 -05:00
c071230a49
Merge pull request #75 from damian0815/sample_generation_refactor_redo
...
Refactor sample generation and introduce sample_prompts.json
2023-02-18 14:42:55 -05:00
230cab9e27
isolate RNG in sample generation
2023-02-18 20:18:21 +01:00
2f48460691
fix ckpt save if only 1 step per epoch, revert epsilon for amp
2023-02-18 11:23:02 -05:00
648fe20200
better default batch size handling
2023-02-18 15:55:54 +01:00
e97f0816db
Squashed commit of the following:
...
commit 86fa1363852850e87be11e5a277b71435f6a3451
Author: Damian Stewart <d@damianstewart.com>
Date: Sat Feb 18 14:43:57 2023 +0100
cleanup, add back random caption support
commit f9a10842b47b9a5d51d53de8d56cb7089a1eeeb2
Author: Damian Stewart <d@damianstewart.com>
Date: Sat Feb 18 13:52:22 2023 +0100
misc fixes and documentation
commit 46167806892258fef509f14e9d83ceab08725cd6
Author: Damian Stewart <d@damianstewart.com>
Date: Sat Feb 18 12:11:18 2023 +0100
works
commit 390bcdf4d8165315e2f84404c62b410c7b674c84
Author: Damian Stewart <d@damianstewart.com>
Date: Sat Feb 18 10:12:14 2023 +0100
SampleGenerator code in place (untested)
commit 022724fa7a435371081fd489ee7e5dbfc2df37ec
Author: Damian Stewart <d@damianstewart.com>
Date: Sat Feb 18 10:17:05 2023 +0100
cleanup and new approach (untested)
commit 4ac81f0924146a7ac3c46f4a4382e7dceaaac47c
Author: Damian Stewart <d@damianstewart.com>
Date: Fri Jan 27 17:26:12 2023 +0100
fix 'classmethod is not callable' error
commit c875933096464a867a5c3cfbf9592605f201f79e
Author: Damian Stewart <d@damianstewart.com>
Date: Fri Jan 27 17:10:03 2023 +0100
fix prompts log crash
commit 2771d52485191388dfa5b3b8892ed7327d874ed6
Author: Damian Stewart <d@damianstewart.com>
Date: Fri Jan 27 14:38:39 2023 +0100
fix circular import
commit 8452272b02fe64a2345fba067a55e51c52debd98
Author: Damian Stewart <d@damianstewart.com>
Date: Fri Jan 27 14:33:26 2023 +0100
refactor sample generation (untested)
2023-02-18 15:51:50 +01:00
37cf437a5f
zero frequency noise option to improve contrast
2023-02-15 18:53:08 -05:00
e28b23851d
save_ckpts_from_n_epochs arg to skip early saving of ckpts
2023-02-13 12:18:33 -05:00
2353c4c16a
validation doc update
2023-02-08 16:05:19 -05:00
8b13c7ed1f
val...
2023-02-08 13:34:49 -05:00
27b66ab54f
ability to not do val
2023-02-08 13:23:19 -05:00
8a8a4cf3df
make val optional, revert multiply algo
2023-02-08 13:04:12 -05:00
19347bcaa8
make fractional multiplier logic apply per-directory
2023-02-08 14:15:54 +01:00
a7b00e9ef3
fix multiplier logic
2023-02-08 13:46:58 +01:00
4e37200dda
fix multiplier issues with validation and refactor validation logic
2023-02-08 11:28:45 +01:00
00ae25ae60
Merge branch 'main' into feat_cli_args_override_json_file
2023-02-07 20:31:01 -05:00
6f4bfdc557
temporarily disable val, issues
2023-02-07 18:43:16 -05:00
29396ec21b
update EveryDreamValidator for noprompt's changes
2023-02-07 17:46:30 +01:00
e0ca75cc96
remove redundant update_old_args() function
2023-02-07 13:46:19 +01:00
21a64c38f2
move collate_fn to top level to possibly fix windows issue
2023-02-06 19:03:33 +01:00
c270dbf6a8
ensure dataloader workers exit cleanly on ctrl-c
2023-02-06 19:03:33 +01:00
4b5654452c
more workers
2023-02-06 19:03:33 +01:00
86a5004098
better main-thread detection
2023-02-06 19:03:33 +01:00
50a71e63b6
background load images for a 40% performance improvement
2023-02-06 19:03:33 +01:00
927880e1fc
allow cli args to override config values
2023-02-06 08:16:08 +01:00
d99b3b1d9b
Merge pull request #38 from noprompt/push-dlma-into-main
...
Push DLMA into `main`, improvements to `data.resolve`
2023-02-05 08:20:10 -05:00
c8c658d181
Forgot to pass args to write_batch_schedule
2023-01-29 18:28:07 -08:00
f96d44ddb4
Move image resolution into its own function
2023-01-29 18:20:40 -08:00
56f130c027
Forgot to add train.py earlier 🤦 ; move write_batch_schedule to train.py
2023-01-29 18:11:34 -08:00
3aa9139b4b
re-read sample prompts every time they're generated
2023-01-30 00:51:03 +01:00
bc273d0512
change training example to constant lr
2023-01-28 18:20:04 -05:00
94feea99ee
fix colab notebook
2023-01-25 20:55:24 -05:00
448848d5c2
fix `'int' object is not callable` warning
2023-01-25 11:06:52 +01:00
4433bd7806
Merge pull request #19 from janekm/wandb_fix
...
Fix wandb setup
2023-01-23 17:38:49 -08:00
067ea506a2
check for local files before downloading from HF
2023-01-23 21:10:04 +01:00
7a78ad0dfa
prevent a crash if hf_repo_subfolder is unset
2023-01-23 19:38:47 +01:00
4f3b4d7dda
clarify code comments
2023-01-23 19:20:37 +01:00
d24dd681c0
Merge remote-tracking branch 'upstream/main' into hf_model_download
2023-01-23 19:19:22 +01:00
9f5f773c33
remove saving yaml for sd1x models, unneeded
2023-01-22 21:43:03 -05:00
18d1da0459
bug fix and multiplytxt fraction stuff
2023-01-22 18:59:59 -05:00
24b00ab35b
add fractional support for multiply.txt
2023-01-22 01:15:50 -05:00
ba5706c5d8
Fix wandb setup
...
Still need two have a log_writer
2023-01-21 23:26:04 +00:00
Victor Hall