Commit 429442f4 authored by AUTOMATIC's avatar AUTOMATIC

fix iterator bug for #2295

parent 8561d576
...@@ -323,10 +323,10 @@ class FrozenCLIPEmbedderWithCustomWords(torch.nn.Module): ...@@ -323,10 +323,10 @@ class FrozenCLIPEmbedderWithCustomWords(torch.nn.Module):
tokens = [] tokens = []
multipliers = [] multipliers = []
for i in range(len(remade_batch_tokens)): for j in range(len(remade_batch_tokens)):
if len(remade_batch_tokens[i]) > 0: if len(remade_batch_tokens[j]) > 0:
tokens.append(remade_batch_tokens[i][:75]) tokens.append(remade_batch_tokens[j][:75])
multipliers.append(batch_multipliers[i][:75]) multipliers.append(batch_multipliers[j][:75])
else: else:
tokens.append([self.wrapped.tokenizer.eos_token_id] * 75) tokens.append([self.wrapped.tokenizer.eos_token_id] * 75)
multipliers.append([1.0] * 75) multipliers.append([1.0] * 75)
......
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