From 9a7f1e7978e2cab7a62bae9760d751d74c1954e1 Mon Sep 17 00:00:00 2001 From: awsr <43862868+awsr@users.noreply.github.com> Date: Fri, 13 Mar 2026 00:15:54 -0700 Subject: [PATCH] Typing update --- modules/ui_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/ui_common.py b/modules/ui_common.py index a1d4481c3..c771a0ad9 100644 --- a/modules/ui_common.py +++ b/modules/ui_common.py @@ -459,7 +459,7 @@ def update_token_counter(text: str): try: try: - ids = getattr(tokenizer(prompt_list), 'input_ids', []) + ids: list = getattr(tokenizer(prompt_list), 'input_ids', []) except TypeError: for p in prompt_list: ids.append(getattr(tokenizer(p), 'input_ids', []))