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 bug in web ui

Browse Source
This commit is contained in:
hiyouga
2023-11-16 15:21:24 +08:00
parent 7537dd434f
commit 6efa38be46
3 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -236,11 +236,12 @@ class Runner:
yield from self._launch(data, do_train=False)
def monitor(self) -> Generator[Tuple[str, Dict[str, Any]], None, None]:
get = lambda name: self.running_data[self.manager.get_elem_by_name(name)]
self.running = True
lang = self.running_data[self.manager.get_elem_by_name("top.lang")]
output_dir = self.running_data[self.manager.get_elem_by_name(
lang = get("top.lang")
output_dir = get_save_dir(get("top.model_name"), get("top.finetuning_type"), get(
"{}.output_dir".format("train" if self.do_train else "eval")
)]
))
while self.thread.is_alive():
time.sleep(2)
if self.aborted: