Add option to disable prompt token counters

1 files changed, 8 insertions, 0 deletions

diff --git a/javascript/token-counters.js b/javascript/token-counters.js
index 0b74658c..9d81a723 100644
--- a/javascript/token-counters.js
+++ b/javascript/token-counters.js
@@ -21,6 +21,9 @@ function update_img2img_tokens(...args) {
 }
 
 function update_token_counter(button_id) {
+    if (opts.disable_token_counters) {
+        return;
+    }
     if (promptTokenCountTimeouts[button_id]) {
         clearTimeout(promptTokenCountTimeouts[button_id]);
     }
@@ -54,6 +57,11 @@ function setupTokenCounting(id, id_counter, id_button) {
     var counter = gradioApp().getElementById(id_counter);
     var textarea = gradioApp().querySelector(`#${id} > label > textarea`);
 
+    if (opts.disable_token_counters) {
+        counter.style.display = "none";
+        return;
+    }
+
     if (counter.parentElement == prompt.parentElement) {
         return;
     }