From 047e3bb4e9ca3ef11bdbbfc7bf1b554dda99026d Mon Sep 17 00:00:00 2001 From: "Dave Lage (rockerBOO)" Date: Thu, 12 Jan 2023 22:11:07 -0500 Subject: [PATCH 1/2] Add optimizer to save_embedding Optimizer was added in bddebe09edeb6a18f2c06986d5658a7be3a563ea. --- scripts/embedding_editor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/embedding_editor.py b/scripts/embedding_editor.py index b000dde..964c956 100644 --- a/scripts/embedding_editor.py +++ b/scripts/embedding_editor.py @@ -248,7 +248,9 @@ def save_embedding_weights(embedding_name, vector_num, *weights): checkpoint = sd_models.select_checkpoint() filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt') - save_embedding(embedding, checkpoint, embedding_name, filename, remove_cached_checksum=True) + optimizer = torch.optim.AdamW([embedding.vec]) + + save_embedding(embedding, optimizer, checkpoint, embedding_name, filename, remove_cached_checksum=True) def update_guidance_embeddings(text): try: From 8123baed7e4b8e1ec546bc7aa25dc19e4fd13a69 Mon Sep 17 00:00:00 2001 From: "Dave Lage (rockerBOO)" Date: Thu, 12 Jan 2023 22:30:04 -0500 Subject: [PATCH 2/2] Add print informing of saved embedding --- scripts/embedding_editor.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/embedding_editor.py b/scripts/embedding_editor.py index 964c956..b67c810 100644 --- a/scripts/embedding_editor.py +++ b/scripts/embedding_editor.py @@ -252,6 +252,8 @@ def save_embedding_weights(embedding_name, vector_num, *weights): save_embedding(embedding, optimizer, checkpoint, embedding_name, filename, remove_cached_checksum=True) + print(f"Saved embedding to {filename}") + def update_guidance_embeddings(text): try: cond_model = shared.sd_model.cond_stage_model