Commit 608fcc25 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix bug with weights when prompt is long.

parent ddc6f12a
...@@ -23,11 +23,11 @@ class ClipTokenWeightEncoder: ...@@ -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)
......
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