aboutsummaryrefslogtreecommitdiffstats
path: root/modules/textual_inversion/textual_inversion.py
AgeCommit message (Collapse)AuthorLines
2022-12-03Merge branch 'master' into racecond_fixAUTOMATIC1111-148/+186
2022-11-30Use devices.autocast instead of torch.autocastbrkirch-1/+1
2022-11-27Merge remote-tracking branch 'flamelaw/master'AUTOMATIC-141/+182
2022-11-27set TI AdamW default weight decay to 0flamelaw-1/+1
2022-11-26Add support Stable Diffusion 2.0AUTOMATIC-4/+3
2022-11-23small fixesflamelaw-1/+1
2022-11-21fix pin_memory with different latent sampling methodflamelaw-6/+1
2022-11-20Gradient accumulation, autocast fix, new latent sampling method, etcflamelaw-137/+183
2022-11-19change StableDiffusionProcessing to internally use sampler name instead of ↵AUTOMATIC-2/+2
sampler index
2022-11-04Fixes race condition in training when VAE is unloadedFampai-0/+5
set_current_image can attempt to use the VAE when it is unloaded to the CPU while training
2022-10-31Fixed minor bugFampai-0/+1
when unloading vae during TI training, generating images after training will error out
2022-10-31Merge branch 'master' of ↵Fampai-25/+62
https://github.com/AUTOMATIC1111/stable-diffusion-webui into TI_optimizations
2022-10-31Added TI training optimizationsFampai-0/+9
option to use xattention optimizations when training option to unload vae when training
2022-10-30Fix dataset still being loaded even when training will be skippedMuhammad Rizqi Nur-1/+1
2022-10-30Add missing info on hypernetwork/embedding model logMuhammad Rizqi Nur-13/+26
Mentioned here: https://github.com/AUTOMATIC1111/stable-diffusion-webui/discussions/1528#discussioncomment-3991513 Also group the saving into one
2022-10-30Revert "Add cleanup after training"Muhammad Rizqi Nur-95/+90
This reverts commit 3ce2bfdf95bd5f26d0f6e250e67338ada91980d1.
2022-10-29Add cleanup after trainingMuhammad Rizqi Nur-90/+95
2022-10-29Add input validations before loading dataset for trainingMuhammad Rizqi Nur-12/+36
2022-10-28Fix log off by 1Muhammad Rizqi Nur-12/+12
2022-10-26typo: cmd_opts.embedding_dir to cmd_opts.embeddings_dirDepFA-1/+1
2022-10-26Implement PR #3625 but for embeddings.timntorres-1/+1
2022-10-26Implement PR #3309 but for embeddings.timntorres-1/+8
2022-10-26Implement PR #3189 but for embeddings.timntorres-5/+5
2022-10-26enable creating embedding with --medvramAUTOMATIC-0/+3
2022-10-21Merge branch 'ae'AUTOMATIC-0/+1
2022-10-20allow overwrite old embeddingDepFA-2/+3
2022-10-18Merge branch 'master' into test_resolve_conflictsMalumaDev-0/+1
2022-10-17print list of embeddings on reloadDepFA-0/+1
2022-10-16Merge branch 'master' into test_resolve_conflictsMalumaDev-0/+1
2022-10-16do not reload embeddings from disk when doing textual inversionAUTOMATIC-0/+1
2022-10-16Merge branch 'master' into test_resolve_conflictsMalumaDev-4/+13
2022-10-15Change vector size footer labelDepFA-1/+1
2022-10-15generalise to popular lossless formatsDepFA-6/+2
2022-10-15add webp for embed loadDepFA-0/+4
2022-10-15only save 1 image per embeddingDepFA-1/+4
2022-10-15add vector size to embed textDepFA-1/+7
2022-10-15Merge branch 'master' into test_resolve_conflictsMalumaDev-25/+54
2022-10-15add option to use batch size for trainingAUTOMATIC-9/+8
2022-10-14remove duplicate code for log loss, add step, make it read from options ↵AUTOMATIC-14/+30
rather than gradio input
2022-10-14Merge remote-tracking branch 'Melanpan/master'AUTOMATIC-0/+16
2022-10-14add option to read generation params for learning previews from txt2imgAUTOMATIC-7/+18
2022-10-14initMalumaDev-10/+25
2022-10-13Add learn_rate to csv and removed a left-over debug statementMelan-2/+3
2022-10-12Save a csv containing the loss while trainingMelan-1/+16
2022-10-12train: change filename processing to be more simple and configurableAUTOMATIC-22/+13
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
2022-10-12Merge pull request #2037 from AUTOMATIC1111/embed-embeddings-in-imagesAUTOMATIC1111-2/+44
Add option to store TI embeddings in png chunks, and load from same.
2022-10-12formattingDepFA-11/+11
2022-10-11spacingDepFA-1/+1
2022-10-11update importsDepFA-0/+3
2022-10-11correct conflict resolution typoDepFA-1/+1