hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

switched the token counter to use hidden buttons instead of api call

Browse Source
This commit is contained in:
Liam
2022-09-27 19:29:53 -04:00
parent 981fe9c4a3
commit e5707b66d6
4 changed files with 25 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/sd_hijack.py
View File

@@ -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):