From 6686a628aa942b1ae36f8a5cbad67f0cbb0ed98b Mon Sep 17 00:00:00 2001
From: Ramon Calvo <ramon@ramoncalvo.com>
Date: Thu, 31 Oct 2024 13:32:19 +0100
Subject: [PATCH] cleanup: remove print

---
 src/tiny_ddpm/sample.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/tiny_ddpm/sample.py b/src/tiny_ddpm/sample.py
index 8528151..fd1cc24 100644
--- a/src/tiny_ddpm/sample.py
+++ b/src/tiny_ddpm/sample.py
@@ -123,7 +123,6 @@ def ddim_sample_images(
         t = torch.full((batch_size,), timesteps[i], device=device, dtype=torch.long)
         x_before = x.clone()
         x = ddim_sample(model, x, t, params)
-        print(f"Step {i}, max diff: {(x - x_before).abs().max().item()}")
 
         if x.isnan().any():
             raise ValueError(f"NaN detected at timestep {timesteps[i]}")