fix bug in web demo

2025-12-16 11:50:35 +08:00 · 2023-06-05 17:58:29 +08:00
parent 56eb99106a
commit a38d57ddd7
2 changed files with 7 additions and 4 deletions
--- a/src/utils/peft_trainer.py
+++ b/src/utils/peft_trainer.py
@@ -49,6 +49,8 @@ class LogCallback(TrainerCallback):
        r"""
        Event called after logging the last logs.
        """
+        if "loss" not in state.log_history[-1]:
+            return
        cur_time = time.time()
        cur_steps = state.log_history[-1].get("step")
        elapsed_time = cur_time - self.start_time