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

Update custom_samplers.py

Browse Source
This commit is contained in:
Extraltodeus
2024-08-27 10:37:56 +02:00
committed by GitHub
parent 37bb0861b4
commit d22cbd3351
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_samplers.py
+1 -1
View File
@@ -4,7 +4,7 @@ import comfy.model_patcher
import comfy.samplers import comfy.samplers
@torch.no_grad() @torch.no_grad()
def fast_distance_weights(t,p=2): def fast_distance_weights(t,p):
d = torch.zeros_like(t,device=t.device) d = torch.zeros_like(t,device=t.device)
for i in range(t.shape[0]): for i in range(t.shape[0]):
d[i] = (t - t[i]).abs().sum(dim=0) d[i] = (t - t[i]).abs().sum(dim=0)