This commit is contained in:
hiyouga
2024-02-29 21:03:48 +08:00
parent 1bfa70ce8e
commit 396fd47947

View File

@@ -38,8 +38,8 @@ class Engine:
yield self._form_dict(init_dict) yield self._form_dict(init_dict)
if not self.pure_chat and not self.demo_mode: if not self.pure_chat:
if self.runner.alive: if self.runner.alive and not self.demo_mode:
yield {elem: gr.update(value=value) for elem, value in self.runner.running_data.items()} yield {elem: gr.update(value=value) for elem, value in self.runner.running_data.items()}
if self.runner.do_train: if self.runner.do_train:
yield self._form_dict({"train.resume_btn": {"value": True}}) yield self._form_dict({"train.resume_btn": {"value": True}})