From 6ee8f0610c3adca5be55db2a6f59d987388e3823 Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Tue, 27 Aug 2024 22:59:43 +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 8533a51..177bf16 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 + 1)
+                        d = fast_distance_weights(torch.stack(x_n), re_step + 2)
                         x_n.append(d)
                 x = x + d * dt
         return x