From ab1f7e03263e8af5fc861389aa550d4eeaab3941 Mon Sep 17 00:00:00 2001
From: Xuan Son Nguyen <son@huggingface.co>
Date: Wed, 11 Dec 2024 20:32:24 +0100
Subject: [PATCH] only extract timings when it's enabled

---
 examples/server/webui/index.html  | 2 +-
 examples/server/webui/src/main.js | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/server/webui/index.html b/examples/server/webui/index.html
index 84f86ef91..454b06787 100644
--- a/examples/server/webui/index.html
+++ b/examples/server/webui/index.html
@@ -252,7 +252,7 @@
           <vue-markdown v-else :source="msg.content"></vue-markdown>
           <!-- render timings if enabled -->
           <div class="dropdown dropdown-hover dropdown-top mt-2" v-if="timings && config.showTokensPerSecond">
-            <div tabindex="0" role="button" class="cursor-pointer font-semibold text-sm opacity-70">Speed: {{ timings.predicted_per_second.toFixed(1) }} t/s</div>
+            <div tabindex="0" role="button" class="cursor-pointer font-semibold text-sm opacity-60">Speed: {{ timings.predicted_per_second.toFixed(1) }} t/s</div>
             <div class="dropdown-content bg-base-100 z-10 w-64 p-2 shadow mt-4">
               <b>Prompt</b><br/>
               - Tokens: {{ timings.prompt_n }}<br/>
diff --git a/examples/server/webui/src/main.js b/examples/server/webui/src/main.js
index 94a428eb3..f1f35481d 100644
--- a/examples/server/webui/src/main.js
+++ b/examples/server/webui/src/main.js
@@ -416,7 +416,7 @@ const mainApp = createApp({
             };
           }
           const timings = chunk.timings;
-          if (timings) {
+          if (timings && this.config.showTokensPerSecond) {
             // only extract what's really needed, to save some space
             this.pendingMsg.timings = {
               prompt_n: timings.prompt_n,