From 847957d4d74dfa2723adb3e3b78c5aa5cc907a94 Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Tue, 27 Aug 2024 19:01:38 +0200
Subject: [PATCH] Update custom_samplers.py

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

diff --git a/custom_samplers.py b/custom_samplers.py
index 517ed2c..bea74ba 100644
--- a/custom_samplers.py
+++ b/custom_samplers.py
@@ -70,6 +70,7 @@ def distance_wrap(resample,resample_end=-1,cfgpp=False):
                         d = (new_d + d) / 2
                     else:
                         d = fast_distance_weights(torch.stack(x_n), re_step + 2)
+                        x_n.append(d)
                 x = x + d * dt
         return x
     return sample_distance_advanced