ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-01 20:23:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix eval in webui

Browse Source 
Former-commit-id: 774ef2bf5823d68b9cc254a676f5adb4af533d75
This commit is contained in:
hiyouga
2024-05-04 00:19:19 +08:00
parent 7a4a6a5522
commit 182b974786
6 changed files with 84 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -299,12 +299,12 @@ class Runner:
progress_bar: gr.Slider(visible=False),
}
else:
running_log, running_progress, running_loss = get_trainer_info(output_path)
running_log, running_progress, running_loss = get_trainer_info(output_path, self.do_train)
return_dict = {
output_box: running_log,
progress_bar: running_progress,
}
if self.do_train and running_loss is not None:
if running_loss is not None:
return_dict[loss_viewer] = running_loss
yield return_dict