ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-15 07:54:37 +00:00
commit bbd715e2f4
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/studio/run_real_mlx_smoke.py
View file

@ -296,13 +296,11 @@ def cmd_train(args) -> int:
args = config,
)
def _on_step(step, total, loss, lr, tok_s, peak_gb, elapsed,
num_tokens, grad_norm = None):
def _on_step(
step, total, loss, lr, tok_s, peak_gb, elapsed, num_tokens, grad_norm = None
):
losses_per_step.append(round(float(loss), 4))
grad_text = (
f" grad={grad_norm:.4f}"
if grad_norm is not None else ""
)
grad_text = f" grad={grad_norm:.4f}" if grad_norm is not None else ""
print(
f" step {step}/{total} loss={loss:.4f} lr={lr:.2e} "
f"tok/s={tok_s:.0f} peak={peak_gb:.2f}GB{grad_text}",