From d22cbd3351d64f99f1b35cb107a51b778e8fb7bc Mon Sep 17 00:00:00 2001 From: Extraltodeus Date: Tue, 27 Aug 2024 10:37:56 +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 bbe84be..c7ac537 100644 --- a/custom_samplers.py +++ b/custom_samplers.py @@ -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)