fix callback

Former-commit-id: 22d9a9c2af
2025-12-20 13:50:35 +08:00 · 2023-07-15 17:18:16 +08:00
parent a696148d6b
commit 70b5232f9a
4 changed files with 9 additions and 5 deletions
--- a/src/llmtuner/extras/callbacks.py
+++ b/src/llmtuner/extras/callbacks.py
@@ -46,7 +46,7 @@ class LogCallback(TrainerCallback):
        r"""
        Event called after logging the last logs.
        """
-        if "current_steps" not in state.log_history[-1]:
+        if "step" not in state.log_history[-1]:
            return
        cur_time = time.time()
        cur_steps = state.log_history[-1].get("step")