Fix token max length

2022-09-25 14:13:03 +08:00 · 2022-09-25 14:13:03 +08:00 · 615b2fc9ce
parent b8eae5de93
commit 615b2fc9ce
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@ -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 = {}