From 23cf8ca7c52ef2abb86c820ee751bbafe4d3e6ed Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 8 Jun 2023 23:48:14 -0400
Subject: [PATCH] Fix bug when embedding gets ignored because of mismatched
 size.

---
 comfy/sd1_clip.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/sd1_clip.py b/comfy/sd1_clip.py
index b1a392736..91fb4ff27 100644
--- a/comfy/sd1_clip.py
+++ b/comfy/sd1_clip.py
@@ -82,6 +82,8 @@ class SD1ClipModel(torch.nn.Module, ClipTokenWeightEncoder):
                         next_new_token += 1
                     else:
                         print("WARNING: shape mismatch when trying to apply embedding, embedding will be ignored", y.shape[0], current_embeds.weight.shape[1])
+            while len(tokens_temp) < len(x):
+                tokens_temp += [self.empty_tokens[0][-1]]
             out_tokens += [tokens_temp]
 
         if len(embedding_weights) > 0: