Fix token max length
This commit is contained in:
parent
b8eae5de93
commit
615b2fc9ce
1 changed files with 1 additions and 1 deletions
|
@ -300,7 +300,7 @@ class FrozenCLIPEmbedderWithCustomWords(torch.nn.Module):
|
|||
remade_batch_tokens = []
|
||||
id_start = self.wrapped.tokenizer.bos_token_id
|
||||
id_end = self.wrapped.tokenizer.eos_token_id
|
||||
maxlen = self.wrapped.max_length - 2
|
||||
maxlen = self.wrapped.max_length
|
||||
used_custom_terms = []
|
||||
|
||||
cache = {}
|
||||
|
|
Loading…
Reference in a new issue