423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 19:30:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix eval resuming in webui

Browse Source
This commit is contained in:
hiyouga
2023-10-15 15:45:38 +08:00
parent 3ad8c92eca
commit 273745f9b9
5 changed files with 27 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/llmtuner/webui/engine.py
View File

@@ -39,9 +39,12 @@ class Engine:
yield self._form_dict(resume_dict)
if self.runner.alive: # TODO: resume eval
if self.runner.alive:
yield {elem: gr.update(value=value) for elem, value in self.runner.data.items()}
resume_dict = {"train.resume_btn": {"value": True}}
if self.runner.do_train:
resume_dict = {"train.resume_btn": {"value": True}}
else:
resume_dict = {"eval.resume_btn": {"value": True}}
else:
resume_dict = {"train.output_dir": {"value": get_time()}}
yield self._form_dict(resume_dict)