preprocessing for textual inversion added

1 files changed, 11 insertions, 3 deletions

diff --git a/modules/textual_inversion/ui.py b/modules/textual_inversion/ui.py
index 633037d8..f19ac5e0 100644
--- a/modules/textual_inversion/ui.py
+++ b/modules/textual_inversion/ui.py
@@ -2,24 +2,31 @@ import html
 

 import gradio as gr

 

-import modules.textual_inversion.textual_inversion as ti

+import modules.textual_inversion.textual_inversion

+import modules.textual_inversion.preprocess

 from modules import sd_hijack, shared

 

 

 def create_embedding(name, initialization_text, nvpt):

-    filename = ti.create_embedding(name, nvpt, init_text=initialization_text)

+    filename = modules.textual_inversion.textual_inversion.create_embedding(name, nvpt, init_text=initialization_text)

 

     sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings()

 

     return gr.Dropdown.update(choices=sorted(sd_hijack.model_hijack.embedding_db.word_embeddings.keys())), f"Created: {filename}", ""

 

 

+def preprocess(*args):

+    modules.textual_inversion.preprocess.preprocess(*args)

+

+    return "Preprocessing finished.", ""

+

+

 def train_embedding(*args):

 

     try:

         sd_hijack.undo_optimizations()

 

-        embedding, filename = ti.train_embedding(*args)

+        embedding, filename = modules.textual_inversion.textual_inversion.train_embedding(*args)

 

         res = f"""

 Training {'interrupted' if shared.state.interrupted else 'finished'} at {embedding.step} steps.

@@ -30,3 +37,4 @@ Embedding saved to {html.escape(filename)}
         raise

     finally:

         sd_hijack.apply_optimizations()

+