1580158
/
sd-webui-prompt-all-in-one
mirror of https://github.com/Physton/sd-webui-prompt-all-in-one
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #368 from neon-aiart/fix/token-counter-nonetype 

FIX: Token Counter TypeError (v2: Handle cond_stage_model is None)
pull/369/head
Physton 2025-11-15 22:58:33 +08:00 committed by GitHub
parent 6a4fb7015b b49e275eb0
commit 1b2cce5433
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/physton_prompt/get_token_counter.py
View File

@ -4,8 +4,10 @@ from functools import partial, reduce
def get_token_counter(text, steps): def get_token_counter(text, steps):
# Check if the model is fully loaded to prevent TypeError during model switching # Check if the model is fully loaded to prevent TypeError during model switching.
if sd_models.model_data.sd_model is None: # Checks both sd_model and its subcomponent (cond_stage_model).
if sd_models.model_data.sd_model is None or \
sd_models.model_data.sd_model.cond_stage_model is None:
return {"token_count": 0, "max_length": 0} return {"token_count": 0, "max_length": 0}
# copy from modules.ui.py # copy from modules.ui.py