Fix token max length

pyeden · Sep 25, 2022 · 615b2fc · 615b2fc
1 parent b8eae5d
commit 615b2fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
@@ -300,7 +300,7 @@ def forward(self, text):
         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 = {}