Update custom_samplers.py

2025-04-27 23:18:11 +02:00
parent f78a9c1300
commit 4f608c6577
1 changed files with 1 additions and 1 deletions
@@ -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)