From 1bfa70ce8eca408fb15e63fd60422aa7b7233d12 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Thu, 29 Feb 2024 20:09:09 +0800 Subject: [PATCH] fix webui --- src/llmtuner/webui/engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/webui/engine.py b/src/llmtuner/webui/engine.py index 6c87a255..65f69204 100644 --- a/src/llmtuner/webui/engine.py +++ b/src/llmtuner/webui/engine.py @@ -38,7 +38,7 @@ class Engine: yield self._form_dict(init_dict) - if not self.pure_chat: + if not self.pure_chat and not self.demo_mode: if self.runner.alive: yield {elem: gr.update(value=value) for elem, value in self.runner.running_data.items()} if self.runner.do_train: