Update custom_samplers.py

2024-08-27 19:01:38 +02:00
parent 2ddd6d277b
commit 847957d4d7
1 changed files with 1 additions and 0 deletions
@@ -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