diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-11 14:15:09 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-11 14:15:09 +0000 |
| commit | 1eaad955330bbe2d55f6b528c902758739413dc8 (patch) | |
| tree | fba92d854e283d3a413b1b36682bb23171a086d7 /modules/textual_inversion/textual_inversion.py | |
| parent | 7aa8fcac1e45c3ad9c6a40df0e44a346afcd5032 (diff) | |
| parent | e0ee5bf703996b33e6d97aa36e0973ceedc88503 (diff) | |
| download | stable-diffusion-webui-gfx803-1eaad955330bbe2d55f6b528c902758739413dc8.tar.gz stable-diffusion-webui-gfx803-1eaad955330bbe2d55f6b528c902758739413dc8.tar.bz2 stable-diffusion-webui-gfx803-1eaad955330bbe2d55f6b528c902758739413dc8.zip | |
Merge branch 'master' into embed-embeddings-in-images
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 13416a08..8c66aeb5 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -281,7 +281,8 @@ def create_embedding(name, num_vectors_per_token, init_text='*'): return fn
-def train_embedding(embedding_name, learn_rate, data_root, log_directory, training_width, training_height, steps, num_repeats, create_image_every, save_embedding_every, template_file, save_image_with_stored_embedding):
+
+def train_embedding(embedding_name, learn_rate, data_root, log_directory, training_width, training_height, steps, num_repeats, create_image_every, save_embedding_every, template_file, save_image_with_stored_embedding, preview_image_prompt)
assert embedding_name, 'embedding not selected'
shared.state.textinfo = "Initializing textual inversion training..."
@@ -369,12 +370,14 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini if embedding.step > 0 and images_dir is not None and embedding.step % create_image_every == 0:
last_saved_image = os.path.join(images_dir, f'{embedding_name}-{embedding.step}.png')
+ preview_text = text if preview_image_prompt == "" else preview_image_prompt
+
p = processing.StableDiffusionProcessingTxt2Img(
sd_model=shared.sd_model,
- prompt=text,
+ prompt=preview_text,
steps=20,
- height=training_height,
- width=training_width,
+ height=training_height,
+ width=training_width,
do_not_save_grid=True,
do_not_save_samples=True,
)
@@ -405,7 +408,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini image.save(last_saved_image)
- last_saved_image += f", prompt: {text}"
+ last_saved_image += f", prompt: {preview_text}"
shared.state.job_no = embedding.step
|