From 81bfa46702f35a4fca8e88b7b5402bfdbf3d8f07 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 26 Jul 2023 15:21:50 +0200
Subject: Updated.

---
 candle-wasm-examples/llama2-c/src/app.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'candle-wasm-examples/llama2-c')

diff --git a/candle-wasm-examples/llama2-c/src/app.rs b/candle-wasm-examples/llama2-c/src/app.rs
index a2307471..2ecd771e 100644
--- a/candle-wasm-examples/llama2-c/src/app.rs
+++ b/candle-wasm-examples/llama2-c/src/app.rs
@@ -116,6 +116,7 @@ impl Component for App {
                     let start_time = performance_now();
                     self.current_decode = Some(CurrentDecode { start_time });
                     self.status = "generating...".to_string();
+                    self.n_tokens = 0;
                     self.generated.clear();
                     let temp = *self.temperature.borrow();
                     console_log!("temp: {}", temp);
@@ -159,7 +160,6 @@ impl Component for App {
             }
             Msg::WorkerInMsg(inp) => {
                 self.worker.send(inp);
-                self.n_tokens = 0;
                 true
             }
             Msg::UpdateStatus(status) => {
-- 
cgit v1.2.3