From 5306a71b42d99bb9e61f8597be01233848ab6c80 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Thu, 29 Feb 2024 21:03:48 +0800 Subject: [PATCH] tiny fix Former-commit-id: 59116aa07fa5fc608f8b801dd3c89e53b117033e --- src/llmtuner/webui/engine.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llmtuner/webui/engine.py b/src/llmtuner/webui/engine.py index 65f69204..78a22a21 100644 --- a/src/llmtuner/webui/engine.py +++ b/src/llmtuner/webui/engine.py @@ -38,8 +38,8 @@ class Engine: yield self._form_dict(init_dict) - if not self.pure_chat and not self.demo_mode: - if self.runner.alive: + if not self.pure_chat: + if self.runner.alive and not self.demo_mode: yield {elem: gr.update(value=value) for elem, value in self.runner.running_data.items()} if self.runner.do_train: yield self._form_dict({"train.resume_btn": {"value": True}})