diff options
| author | dan <guaneec@gmail.com> | 2023-01-07 13:07:27 +0000 |
|---|---|---|
| committer | dan <guaneec@gmail.com> | 2023-01-07 18:14:36 +0000 |
| commit | 448b9cedab66e05b5b2800513ca334a769b42aa7 (patch) | |
| tree | 005de335f1db4e341292a494a874f51838394c9b /modules/textual_inversion/textual_inversion.py | |
| parent | 151233399c4b79934bdbb7c12a97eeb6499572fb (diff) | |
| download | stable-diffusion-webui-gfx803-448b9cedab66e05b5b2800513ca334a769b42aa7.tar.gz stable-diffusion-webui-gfx803-448b9cedab66e05b5b2800513ca334a769b42aa7.tar.bz2 stable-diffusion-webui-gfx803-448b9cedab66e05b5b2800513ca334a769b42aa7.zip | |
Allow variable img size
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 45882ed6..9f96d0fd 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -451,8 +451,8 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_ else:
p.prompt = batch.cond_text[0]
p.steps = 20
- p.width = training_width
- p.height = training_height
+ p.width = batch.img_shape[0][0]
+ p.height = batch.img_shape[0][1]
preview_text = p.prompt
|