diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-14 14:50:25 +0100 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-15 17:23:39 +0300 |
| commit | 9e846083b702a498fdb60accd72f075fa26701d9 (patch) | |
| tree | b0a974b51a17309ef400f7a692cb05b664cd5f5b | |
| parent | 703e6d9e4e161d36b9328eefb5200e1c44fb4afd (diff) | |
add vector size to embed text
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index e754747e..6f549d62 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -327,10 +327,16 @@ def train_embedding(embedding_name, learn_rate, batch_size, data_root, log_direc info.add_text("sd-ti-embedding", embedding_to_b64(data))
title = "<{}>".format(data.get('name', '???'))
+
+ try:
+ vectorSize = list(data['string_to_param'].values())[0].shape[0]
+ except Exception as e:
+ vectorSize = '?'
+
checkpoint = sd_models.select_checkpoint()
footer_left = checkpoint.model_name
footer_mid = '[{}]'.format(checkpoint.hash)
- footer_right = '{}'.format(embedding.step)
+ footer_right = 'v{} {}s'.format(vectorSize, embedding.step)
captioned_image = caption_image_overlay(image, title, footer_left, footer_mid, footer_right)
captioned_image = insert_image_data_embed(captioned_image, data)
|