From 4f608c657719e7ab94c65b959be319f716b1b6b6 Mon Sep 17 00:00:00 2001 From: Extraltodeus Date: Sun, 27 Apr 2025 23:18:11 +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 aaf53cc..eda1e10 100644 --- a/custom_samplers.py +++ b/custom_samplers.py @@ -30,7 +30,7 @@ def fast_distance_weights(t, use_softmax=False, use_slerp=False, uncond=None): if uncond != None: uncond = uncond.div(torch.linalg.matrix_norm(uncond, keepdim=True)) - distances += tn.sub(uncond).abs().div(n) + distances += tn.sub(uncond).abs() #.div(n) if use_softmax: distances = distances.mul(n).softmax(dim=0)