Skip to content

Commit

Permalink
Correct spelling 'token_weight_pars_t5' to 'token_weight_pairs_t5' (c…
Browse files Browse the repository at this point in the history
  • Loading branch information
silveroxides authored Aug 4, 2024
1 parent ddb6a9f commit 7afa985
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions comfy/text_encoders/flux.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,9 +52,9 @@ def reset_clip_options(self):

def encode_token_weights(self, token_weight_pairs):
token_weight_pairs_l = token_weight_pairs["l"]
token_weight_pars_t5 = token_weight_pairs["t5xxl"]
token_weight_pairs_t5 = token_weight_pairs["t5xxl"]

t5_out, t5_pooled = self.t5xxl.encode_token_weights(token_weight_pars_t5)
t5_out, t5_pooled = self.t5xxl.encode_token_weights(token_weight_pairs_t5)
l_out, l_pooled = self.clip_l.encode_token_weights(token_weight_pairs_l)
return t5_out, l_pooled

Expand Down
4 changes: 2 additions & 2 deletions comfy/text_encoders/sd3_clip.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ def reset_clip_options(self):
def encode_token_weights(self, token_weight_pairs):
token_weight_pairs_l = token_weight_pairs["l"]
token_weight_pairs_g = token_weight_pairs["g"]
token_weight_pars_t5 = token_weight_pairs["t5xxl"]
token_weight_pairs_t5 = token_weight_pairs["t5xxl"]
lg_out = None
pooled = None
out = None
Expand All @@ -108,7 +108,7 @@ def encode_token_weights(self, token_weight_pairs):
pooled = torch.cat((l_pooled, g_pooled), dim=-1)

if self.t5xxl is not None:
t5_out, t5_pooled = self.t5xxl.encode_token_weights(token_weight_pars_t5)
t5_out, t5_pooled = self.t5xxl.encode_token_weights(token_weight_pairs_t5)
if lg_out is not None:
out = torch.cat([lg_out, t5_out], dim=-2)
else:
Expand Down

0 comments on commit 7afa985

Please sign in to comment.