diff --git a/modules/ui_common.py b/modules/ui_common.py index 99a8a2d22..0231676c6 100644 --- a/modules/ui_common.py +++ b/modules/ui_common.py @@ -458,11 +458,8 @@ def update_token_counter(text: str): max_length = 0 try: - try: - ids = getattr(tokenizer(prompt_list), 'input_ids', []) - except TypeError: - for p in prompt_list: - ids.append(getattr(tokenizer(p), 'input_ids', [])) + for p in prompt_list: + ids.append(getattr(tokenizer(p), 'input_ids', [])) except Exception as e: shared.log.warning("Token counter:", e) return gr.update(value=f"??/{max_length}", visible=True)