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

fix bug in web ui

Browse Source 
Former-commit-id: a598f145ec903dd2b2c984d951b6c450b142ece5
This commit is contained in:
hiyouga
2023-11-16 15:21:24 +08:00
parent df83def566
commit dab9385297
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: