From 460bbae58726c177beddfcddf351f27e205d3fb2 Mon Sep 17 00:00:00 2001
From: hentailord85ez <112723046+hentailord85ez@users.noreply.github.com>
Date: Mon, 10 Oct 2022 16:09:06 +0100
Subject: [PATCH] Pad beginning of textual inversion embedding

---
 modules/sd_hijack.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 8d5c77d89..3a60cd635 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -151,6 +151,11 @@ class FrozenCLIPEmbedderWithCustomWords(torch.nn.Module):
                 else:
                     emb_len = int(embedding.vec.shape[0])
                     iteration = len(remade_tokens) // 75
+                    if (len(remade_tokens) + emb_len) // 75 != iteration:
+                        rem = (75 * (iteration + 1) - len(remade_tokens))
+                        remade_tokens += [id_end] * rem
+                        multipliers += [1.0] * rem
+                        iteration += 1
                     fixes.append((iteration, (len(remade_tokens) % 75, embedding)))
                     remade_tokens += [0] * emb_len
                     multipliers += [weight] * emb_len