890e68aaf7
Fixed minor bug
...
when unloading vae during TI training, generating images after
training will error out
2022-10-31 10:07:12 -04:00
3b0127e698
Merge branch 'master' of https://github.com/AUTOMATIC1111/stable-diffusion-webui into TI_optimizations
2022-10-31 09:54:51 -04:00
006756f9cd
Added TI training optimizations
...
option to use xattention optimizations when training
option to unload vae when training
2022-10-31 07:26:08 -04:00
17a2076f72
Merge pull request #3928 from R-N/validate-before-load
...
Optimize training a little
2022-10-30 09:51:36 +03:00
3d58510f21
Fix dataset still being loaded even when training will be skipped
2022-10-30 00:54:59 +07:00
a07f054c86
Add missing info on hypernetwork/embedding model log
...
Mentioned here: https://github.com/AUTOMATIC1111/stable-diffusion-webui/discussions/1528#discussioncomment-3991513
Also group the saving into one
2022-10-30 00:49:29 +07:00
ab05a74ead
Revert "Add cleanup after training"
...
This reverts commit 3ce2bfdf95
2022-10-30 00:32:02 +07:00
a27d19de2e
Additional assert on dataset
2022-10-29 19:44:05 +07:00
3ce2bfdf95
Add cleanup after training
2022-10-29 19:43:21 +07:00
ab27c111d0
Add input validations before loading dataset for training
2022-10-29 18:09:17 +07:00
ef4c94e1cf
Improve lr schedule error message
2022-10-29 15:42:51 +07:00
a5f3adbdd7
Allow trailing comma in learning rate
2022-10-29 15:37:24 +07:00
810e6a407d
Merge pull request #3858 from R-N/log-csv
...
Fix log off by 1 #3847
2022-10-29 07:55:20 +03:00
9ceef81f77
Fix log off by 1
2022-10-28 20:48:08 +07:00
a0a7024c67
Fix random dataset shuffle on TI
2022-10-28 02:13:48 +09:00
737eb28fac
typo: cmd_opts.embedding_dir to cmd_opts.embeddings_dir
2022-10-26 17:38:08 +03:00
f4e1464217
Implement PR #3625 but for embeddings.
2022-10-26 10:14:35 +03:00
4875a6c217
Implement PR #3309 but for embeddings.
2022-10-26 10:14:35 +03:00
c2dc9bfa89
Implement PR #3189 but for embeddings.
2022-10-26 10:14:35 +03:00
cbb857b675
enable creating embedding with --medvram
2022-10-26 09:44:02 +03:00
df0c5ea29d
update default weights
2022-10-25 17:06:59 -07:00
54f0c14824
download better face detection module dynamically
2022-10-25 16:14:13 -07:00
db8ed5fe5c
Focal crop UI elements
2022-10-25 15:22:29 -07:00
6629446a2f
Merge branch 'master' into focal-point-cropping
2022-10-25 13:22:27 -07:00
3e6c2420c1
improve debug markers, fix algo weighting
2022-10-25 13:13:12 -07:00
1be5933ba2
auto cropping now works with non square crops
2022-10-23 04:11:07 -07:00
f49c08ea56
prevent error spam when processing images without txt files for captions
2022-10-21 18:46:02 +03:00
5e9afa5c8a
Merge branch 'master' into fix/train-preprocess-keep-ratio
2022-10-21 18:36:29 +03:00
306e2ff6ab
Update image_embedding.py
2022-10-21 16:47:37 +03:00
d0ea471b0c
Use opts in textual_inversion image_embedding.py for dynamic fonts
2022-10-21 16:47:37 +03:00
7d6b388d71
Merge branch 'ae'
2022-10-21 13:35:01 +03:00
0c5522ea21
Merge branch 'master' into training-help-text
2022-10-21 09:57:55 +03:00
b69c37d25e
Allow datasets with only 1 image in TI
2022-10-21 09:54:09 +03:00
85dd62c4c7
train: ui: added `Split image threshold` and `Split image overlap ratio` to preprocess
2022-10-20 23:35:01 +09:00
9681419e42
train: fixed preprocess image ratio
2022-10-20 23:32:41 +09:00
0ddaf8d202
improve face detection a lot
2022-10-20 00:34:55 -07:00
858462f719
do caption copy for both flips
2022-10-20 02:57:18 +01:00
59ed744383
face detection algo, configurability, reusability
...
Try to move the crop in the direction of a face if it is present
More internal configuration options for choosing weights of each of the algorithm's findings
Move logic into its module
2022-10-19 17:19:02 -07:00
9b65c4ecf4
pass preprocess_txt_action param
2022-10-20 00:49:23 +01:00
fbcce66601
add existing caption file handling
2022-10-20 00:46:54 +01:00
c3835ec85c
pass overwrite old flag
2022-10-20 00:24:24 +01:00
0087079c2d
allow overwrite old embedding
2022-10-20 00:10:59 +01:00
41e3877be2
fix entropy point calculation
2022-10-19 13:44:59 -07:00
abeec4b630
Add auto focal point cropping to Preprocess images
...
This algorithm plots a bunch of points of interest on the source
image and averages their locations to find a center.
Most points come from OpenCV. One point comes from an
entropy model. OpenCV points account for 50% of the weight and the
entropy based point is the other 50%.
The center of all weighted points is calculated and a bounding box
is drawn as close to centered over that point as possible.
2022-10-19 03:18:26 -07:00
1997ccff13
Merge branch 'master' into test_resolve_conflicts
2022-10-18 08:55:08 +02:00
62edfae257
print list of embeddings on reload
2022-10-17 08:42:17 +03:00
ae0fdad64a
Merge branch 'master' into test_resolve_conflicts
2022-10-16 17:55:58 +02:00
9324cdaa31
ui fix, re organization of the code
2022-10-16 17:53:56 +02:00
0c5fa9a681
do not reload embeddings from disk when doing textual inversion
2022-10-16 09:09:04 +03:00
97ceaa23d0
Merge branch 'master' into test_resolve_conflicts
2022-10-16 00:06:36 +02:00
Fampai