nodes/Extraltodeus-DistanceSampler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

slight quality improvement

Browse Source
This commit is contained in:
Extraltodeus
2025-04-26 01:10:16 +02:00
committed by GitHub
parent 3f16440843
commit 9e84b150ec
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_samplers.py
+3 -5
View File
@@ -26,18 +26,16 @@ def fast_distance_weights(t, use_softmax=False, use_slerp=False, uncond=None):
tn = t.div(norm) tn = t.div(norm)
distances = (tn.unsqueeze(0) - tn.unsqueeze(1)).abs().sum(dim=0) distances = (tn.unsqueeze(0) - tn.unsqueeze(1)).abs().sum(dim=0)
distances = distances.max(dim=0, keepdim=True).values - distances
if uncond != None: if uncond != None:
uncond = uncond.div(torch.linalg.matrix_norm(uncond, keepdim=True)) uncond = uncond.div(torch.linalg.matrix_norm(uncond, keepdim=True))
distances -= tn.sub(uncond).abs() distances += tn.sub(uncond).abs().div(n)
if use_softmax: if use_softmax:
distances = distances.max(dim=0).values - distances
distances = distances.mul(n).softmax(dim=0) distances = distances.mul(n).softmax(dim=0)
else: else:
distances = 1 - (distances - distances.min(dim=0).values) / (distances.max(dim=0).values - distances.min(dim=0).values) distances = distances.div(distances.max(dim=0).values).pow(2)
distances[~torch.isfinite(distances)] = 1
distances = distances.pow(2)
distances = distances / distances.sum(dim=0) distances = distances / distances.sum(dim=0)
if use_slerp: if use_slerp: