From 9408fa5ff72d9ed947f2a7ee6ea1e0863f9a98ed Mon Sep 17 00:00:00 2001
From: gdamms <damguillotin@gmail.com>
Date: Thu, 2 May 2024 16:49:39 +0200
Subject: oups

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 899538f..4048607 100644
--- a/main.py
+++ b/main.py
@@ -295,7 +295,7 @@ if __name__ == '__main__':
 
         for ti in range(DIFFU_STEPS, 0, -1):
             t = torch.tensor([[ti]], device=DEVICE, dtype=torch.float32)
-            x = p_xt_1_xt(model, x, t, vec).sample()
+            x = p_xt_1_xt(model, x, t, vec)
             if ti in ti_plots:
                 ti_plotind = nb_plots - np.where(ti_plots == ti)[0][0]
                 ax = fig.add_subplot(len(n_values), nb_plots, ti_plotind + nb_plots * attempti) 
-- 
cgit v1.3.1