remove/simplify some changes from #6481

2025-08-08 21:30:27 +00:00 · 2023-01-09 22:52:23 +03:00
parent bdd57ad073
commit 43bb5190fc
3 changed files with 8 additions and 12 deletions
--- a/modules/textual_inversion/textual_inversion.py
+++ b/modules/textual_inversion/textual_inversion.py
@@ -492,8 +492,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 = batch.img_shape[0][0]
-                        p.height = batch.img_shape[0][1]
+                        p.width = training_width
+                        p.height = training_height

                    preview_text = p.prompt