From 55b36bb5fa1fde41546ac1a2591e3d82d645d9ec Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 5 Mar 2026 15:56:40 +0200 Subject: [PATCH] Safer --- comfy/ldm/rt_detr/rtdetr_v4.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/rt_detr/rtdetr_v4.py b/comfy/ldm/rt_detr/rtdetr_v4.py index 426d3e6c3..9443761cb 100644 --- a/comfy/ldm/rt_detr/rtdetr_v4.py +++ b/comfy/ldm/rt_detr/rtdetr_v4.py @@ -464,7 +464,7 @@ def weighting_function(reg_max, up, reg_scale): step = (ub1 + 1) ** (2 / (reg_max - 2)) left = [-(step ** i) + 1 for i in range(reg_max // 2 - 1, 0, -1)] right = [ (step ** i) - 1 for i in range(1, reg_max // 2)] - vals = [-ub2] + left + [torch.zeros_like(up[0][None])] + right + [ub2] + vals = [-ub2] + left + [0] + right + [ub2] return torch.tensor(vals, dtype=up.dtype, device=up.device)