"...llama_fastertransformer.git" did not exist on "54961bb0d3eee1c5d4e4e12fd195a264d321ed0b"
Commit 0d972b85 authored by comfyanonymous's avatar comfyanonymous
Browse files

This seems to give better quality in tome.

parent 18a6c1db
...@@ -109,7 +109,7 @@ def get_functions(x, ratio, original_shape): ...@@ -109,7 +109,7 @@ def get_functions(x, ratio, original_shape):
w = original_w // downsample w = original_w // downsample
h = original_h // downsample h = original_h // downsample
r = int(x.shape[1] * ratio) 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) m, u = bipartite_soft_matching_random2d(x, w, h, stride_x, stride_y, r, no_rand)
return m, u return m, u
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment