298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 19:03:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug with weights when prompt is long.

Browse Source
This commit is contained in:
comfyanonymous 2023-07-06 02:43:40 -04:00
parent ddc6f12ad5
commit 608fcc2591
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/sd1_clip.py
View File

@ -23,11 +23,11 @@ class ClipTokenWeightEncoder:
first_pooled = pooled[0:1] first_pooled = pooled[0:1]
output = [] output = []
for i in range(1, out.shape[0]): for k in range(1, out.shape[0]):
z = out[i:i+1] z = out[k:k+1]
for i in range(len(z)): for i in range(len(z)):
for j in range(len(z[i])): for j in range(len(z[i])):
weight = token_weight_pairs[i - 1][j][1] weight = token_weight_pairs[k - 1][j][1]
z[i][j] = (z[i][j] - z_empty[0][j]) * weight + z_empty[0][j] z[i][j] = (z[i][j] - z_empty[0][j]) * weight + z_empty[0][j]
output.append(z) output.append(z)