From 034eb5fb49a8b1dd92f2bfa0a31bdca707c7199b Mon Sep 17 00:00:00 2001 From: levzzz Date: Wed, 18 Feb 2026 21:30:21 +0500 Subject: [PATCH] Fix first step logging, don't commit without ask --- adaptive_sampler.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/adaptive_sampler.py b/adaptive_sampler.py index 417df9a..29ce98b 100644 --- a/adaptive_sampler.py +++ b/adaptive_sampler.py @@ -60,7 +60,10 @@ def sample_adaptive_custom( }) if not disable_pbar: - print(f"Step {total_steps + 1}: sigma={current_sigma:.6f}, step_size={step_size:.6f}, error={error_val:.6f}") + if eps_prev is not None: + print(f"Step {total_steps + 1}: sigma={current_sigma:.6f}, step_size={step_size:.6f}, error={error_val:.6f}") + else: + print(f"Step {total_steps + 1}: sigma={current_sigma:.6f}, step_size={step_size:.6f}") # Get current prediction eps_current = model(x, current_sigma * s_in, **extra_args)