wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-18 23:55:08 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

Safer

Browse Source
This commit is contained in:
kijai 2026-03-05 15:56:40 +02:00
parent 2c760cbf98
commit 55b36bb5fa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/rt_detr/rtdetr_v4.py
View File

@ -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)