diff --git a/modules/ui_common.py b/modules/ui_common.py
index 0231676c6..99a8a2d22 100644
--- a/modules/ui_common.py
+++ b/modules/ui_common.py
@@ -458,8 +458,11 @@ def update_token_counter(text: str):
             max_length = 0
 
         try:
-            for p in prompt_list:
-                ids.append(getattr(tokenizer(p), 'input_ids', []))
+            try:
+                ids = getattr(tokenizer(prompt_list), 'input_ids', [])
+            except TypeError:
+                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"<span class='gr-box gr-text-input'>??/{max_length}</span>", visible=True)