diff --git a/comfy/ldm/modules/tomesd.py b/comfy/ldm/modules/tomesd.py
index 5bf1acec..1eafcd0a 100644
--- a/comfy/ldm/modules/tomesd.py
+++ b/comfy/ldm/modules/tomesd.py
@@ -109,7 +109,7 @@ def get_functions(x, ratio, original_shape):
         w = original_w // downsample
         h = original_h // downsample
         r = int(x.shape[1] * ratio)
-        no_rand = True
+        no_rand = False
         m, u = bipartite_soft_matching_random2d(x, w, h, stride_x, stride_y, r, no_rand)
         return m, u