diff options
| author | Fampai <> | 2022-10-31 14:07:12 +0000 |
|---|---|---|
| committer | Fampai <> | 2022-10-31 14:07:12 +0000 |
| commit | 890e68aaf75ae80d5eb2fa95b4bf1adf78b96881 (patch) | |
| tree | 7d7056afd26b3fb2778d9c4dca4a825cfd565baa /modules/textual_inversion/textual_inversion.py | |
| parent | 3b0127e698a2eeb913437bce0b25b478fb06ff11 (diff) | |
| download | stable-diffusion-webui-gfx803-890e68aaf75ae80d5eb2fa95b4bf1adf78b96881.tar.gz stable-diffusion-webui-gfx803-890e68aaf75ae80d5eb2fa95b4bf1adf78b96881.tar.bz2 stable-diffusion-webui-gfx803-890e68aaf75ae80d5eb2fa95b4bf1adf78b96881.zip | |
Fixed minor bug
when unloading vae during TI training, generating images after
training will error out
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 54a734f1..0aeb0459 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -409,6 +409,7 @@ Last saved image: {html.escape(last_saved_image)}<br/> filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt')
save_embedding(embedding, checkpoint, embedding_name, filename, remove_cached_checksum=True)
+ shared.sd_model.first_stage_model.to(devices.device)
return embedding, filename
|