Update custom_samplers.py

This commit is contained in:
Extraltodeus
2024-08-27 10:37:56 +02:00
committed by GitHub
parent 37bb0861b4
commit d22cbd3351
+1 -1
View File
@@ -4,7 +4,7 @@ import comfy.model_patcher
import comfy.samplers
@torch.no_grad()
def fast_distance_weights(t,p=2):
def fast_distance_weights(t,p):
d = torch.zeros_like(t,device=t.device)
for i in range(t.shape[0]):
d[i] = (t - t[i]).abs().sum(dim=0)