From 599d9cd5bf624c3f9a415a9dab3b45f490524783 Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Tue, 27 Aug 2024 22:54:44 +0200
Subject: [PATCH] Update custom_samplers.py

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

diff --git a/custom_samplers.py b/custom_samplers.py
index 177bf16..8533a51 100644
--- a/custom_samplers.py
+++ b/custom_samplers.py
@@ -69,7 +69,7 @@ def distance_wrap(resample,resample_end=-1,cfgpp=False):
                     if re_step == 0:
                         d = (new_d + d) / 2
                     else:
-                        d = fast_distance_weights(torch.stack(x_n), re_step + 2)
+                        d = fast_distance_weights(torch.stack(x_n), re_step + 1)
                         x_n.append(d)
                 x = x + d * dt
         return x