From e5707b66d6db2c019bfccf66f9ba53e3daaea40b Mon Sep 17 00:00:00 2001
From: Liam <liamthekerr@gmail.com>
Date: Tue, 27 Sep 2022 19:29:53 -0400
Subject: switched the token counter to use hidden buttons instead of api call

---
 modules/sd_hijack.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'modules/sd_hijack.py')

diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 4d799ac0..bfbd07f9 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -273,8 +273,7 @@ class StableDiffusionModelHijack:
     def tokenize(self, text):
         max_length = self.clip.max_length - 2
         _, remade_batch_tokens, _, _, _, token_count = self.clip.process_text([text])
-        return {"tokens": remade_batch_tokens[0], "token_count":token_count, "max_length":max_length}
-        
+        return remade_batch_tokens[0], token_count, max_length
 
 class FrozenCLIPEmbedderWithCustomWords(torch.nn.Module):
     def __init__(self, wrapped, hijack):
-- 
cgit v1.2.1