diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 16:40:51 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 16:40:51 +0000 |
| commit | 88ec0cf5571883d84abd09196652b3679e359f2e (patch) | |
| tree | 030b33b3060c750ea5e8212049c293388f2fc3b3 /modules/textual_inversion/ui.py | |
| parent | 53a3dc601fb734ce433505b1ca68770919106bad (diff) | |
| download | stable-diffusion-webui-gfx803-88ec0cf5571883d84abd09196652b3679e359f2e.tar.gz stable-diffusion-webui-gfx803-88ec0cf5571883d84abd09196652b3679e359f2e.tar.bz2 stable-diffusion-webui-gfx803-88ec0cf5571883d84abd09196652b3679e359f2e.zip | |
fix for incorrect embedding token length calculation (will break seeds that use embeddings, you're welcome!)
add option to input initialization text for embeddings
Diffstat (limited to 'modules/textual_inversion/ui.py')
| -rw-r--r-- | modules/textual_inversion/ui.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/textual_inversion/ui.py b/modules/textual_inversion/ui.py index ce3677a9..66c43ffb 100644 --- a/modules/textual_inversion/ui.py +++ b/modules/textual_inversion/ui.py @@ -6,8 +6,8 @@ import modules.textual_inversion.textual_inversion as ti from modules import sd_hijack, shared
-def create_embedding(name, nvpt):
- filename = ti.create_embedding(name, nvpt)
+def create_embedding(name, initialization_text, nvpt):
+ filename = ti.create_embedding(name, nvpt, init_text=initialization_text)
sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings()
|