make it not break on empty inputs; thank you tarded, we are
This commit is contained in:
parent
79e39fae61
commit
08066676a4
|
@ -147,7 +147,7 @@ class FrozenCLIPEmbedderWithCustomWordsBase(torch.nn.Module):
|
||||||
chunk.multipliers += [weight] * emb_len
|
chunk.multipliers += [weight] * emb_len
|
||||||
position += embedding_length_in_tokens
|
position += embedding_length_in_tokens
|
||||||
|
|
||||||
if len(chunk.tokens) > 0:
|
if len(chunk.tokens) > 0 or len(chunks) == 0:
|
||||||
next_chunk()
|
next_chunk()
|
||||||
|
|
||||||
return chunks, token_count
|
return chunks, token_count
|
||||||
|
|
Loading…
Reference in New Issue