Skip to content

GitLab

"...components/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "403195f0eb4b78559b51611eb2529836cd17badb"
Commit 23cf8ca7 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix bug when embedding gets ignored because of mismatched size.

parent 29c50954
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
comfy/sd1_clip.py
View file @ 23cf8ca7
...@@ -82,6 +82,8 @@ class SD1ClipModel(torch.nn.Module, ClipTokenWeightEncoder): ...@@ -82,6 +82,8 @@ class SD1ClipModel(torch.nn.Module, ClipTokenWeightEncoder):
next_new_token += 1 next_new_token += 1
else: else:
print("WARNING: shape mismatch when trying to apply embedding, embedding will be ignored", y.shape[0], current_embeds.weight.shape[1]) 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] out_tokens += [tokens_temp]
if len(embedding_weights) > 0: if len(embedding_weights) > 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment