3bf5591efe
fix F541 f-string without any placeholders
3 years ago
c0355caefe
Fix various typos
3 years ago
c9a2cfdf2a
Merge branch 'master' into racecond_fix
3 years ago
4d5f1691dd
Use devices.autocast instead of torch.autocast
3 years ago
b48b7999c8
Merge remote-tracking branch 'flamelaw/master'
3 years ago
755df94b2a
set TI AdamW default weight decay to 0
3 years ago
ce6911158b
Add support Stable Diffusion 2.0
3 years ago
89d8ecff09
small fixes
3 years ago
5b57f61ba4
fix pin_memory with different latent sampling method
3 years ago
bd68e35de3
Gradient accumulation, autocast fix, new latent sampling method, etc
3 years ago
cdc8020d13
change StableDiffusionProcessing to internally use sampler name instead of sampler index
3 years ago
39541d7725
Fixes race condition in training when VAE is unloaded
...
set_current_image can attempt to use the VAE when it is unloaded to
the CPU while training
3 years ago
890e68aaf7
Fixed minor bug
...
when unloading vae during TI training, generating images after
training will error out
3 years ago
3b0127e698
Merge branch 'master' of https://github.com/AUTOMATIC1111/stable-diffusion-webui into TI_optimizations
3 years ago
006756f9cd
Added TI training optimizations
...
option to use xattention optimizations when training
option to unload vae when training
3 years ago
3d58510f21
Fix dataset still being loaded even when training will be skipped
3 years ago
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
3 years ago
ab05a74ead
Revert "Add cleanup after training"
...
This reverts commit 3ce2bfdf95
3 years ago
3ce2bfdf95
Add cleanup after training
3 years ago
ab27c111d0
Add input validations before loading dataset for training
3 years ago
9ceef81f77
Fix log off by 1
3 years ago
737eb28fac
typo: cmd_opts.embedding_dir to cmd_opts.embeddings_dir
3 years ago
f4e1464217
Implement PR #3625 but for embeddings.
3 years ago
4875a6c217
Implement PR #3309 but for embeddings.
3 years ago
c2dc9bfa89
Implement PR #3189 but for embeddings.
3 years ago
cbb857b675
enable creating embedding with --medvram
3 years ago
7d6b388d71
Merge branch 'ae'
3 years ago
0087079c2d
allow overwrite old embedding
3 years ago
1997ccff13
Merge branch 'master' into test_resolve_conflicts
3 years ago
62edfae257
print list of embeddings on reload
3 years ago
ae0fdad64a
Merge branch 'master' into test_resolve_conflicts
3 years ago
0c5fa9a681
do not reload embeddings from disk when doing textual inversion
3 years ago
97ceaa23d0
Merge branch 'master' into test_resolve_conflicts
3 years ago
b6e3b96dab
Change vector size footer label
3 years ago
ddf6899df0
generalise to popular lossless formats
3 years ago
9a1dcd78ed
add webp for embed load
3 years ago
939f16529a
only save 1 image per embedding
3 years ago
9e846083b7
add vector size to embed text
3 years ago
7b7561f6e4
Merge branch 'master' into test_resolve_conflicts
3 years ago
c7a86f7fe9
add option to use batch size for training
3 years ago
03d62538ae
remove duplicate code for log loss, add step, make it read from options rather than gradio input
3 years ago
326fe7d44b
Merge remote-tracking branch 'Melanpan/master'
3 years ago
c344ba3b32
add option to read generation params for learning previews from txt2img
3 years ago
bb57f30c2d
init
3 years ago
8636b50aea
Add learn_rate to csv and removed a left-over debug statement
3 years ago
1cfc2a1898
Save a csv containing the loss while training
3 years ago
c3c8eef9fd
train: change filename processing to be more simple and configurable
...
train: make it possible to make text files with prompts
train: rework scheduler so that there's less repeating code in textual inversion and hypernets
train: move epochs setting to options
3 years ago
cc5803603b
Merge pull request #2037 from AUTOMATIC1111/embed-embeddings-in-images
...
Add option to store TI embeddings in png chunks, and load from same.
3 years ago
10a2de644f
formatting
3 years ago
5f3317376b
spacing
3 years ago
Yuval Aboulafia
Jim Hays
AUTOMATIC1111
brkirch
flamelaw
Fampai
Muhammad Rizqi Nur
DepFA
timntorres
MalumaDev
Melan