ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix eval resuming in webui

Browse Source 
Former-commit-id: b28b53cd06777f213ef7b925a914ff5fd357ade1
This commit is contained in:
hiyouga
2023-10-15 15:45:38 +08:00
parent 7070f3969d
commit 68330eab2a
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)