wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-20 05:18:15 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

Reduce UV unwrap VRAM usage

Browse Source
This commit is contained in:
kijai 2026-07-17 14:44:14 +03:00
parent 550af28f45
commit f40e8a099b
1 changed files with 32 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/mesh3d/uv_unwrap/pack.py
View File

@ -468,8 +468,14 @@ def _raster_all_torch(uvs_tex_pad, faces_pad, fmask, bw_t, bh_t, padding, device
d11 = (v1 * v1).sum(-1) d11 = (v1 * v1).sum(-1)
den = (d00 * d11 - d01 * d01).clamp(min=1e-20) den = (d00 * d11 - d01 * d01).clamp(min=1e-20)
free = comfy.model_management.get_free_memory(device)
budget = int(min(1 << 23, max(1 << 20, (free * 0.25) / 56)))
for g in sorted(set(bsz.tolist())): # one batch per pow2 grid for g in sorted(set(bsz.tolist())): # one batch per pow2 grid
sel = (bsz == g).nonzero(as_tuple=True)[0] sel_g = (bsz == g).nonzero(as_tuple=True)[0]
per = max(1, budget // (g * g))
for cs in range(0, sel_g.shape[0], per):
sel = sel_g[cs:cs + per]
m = sel.shape[0] m = sel.shape[0]
xs0 = x0[sel].view(m, 1, 1) xs0 = x0[sel].view(m, 1, 1)
ys0 = y0[sel].view(m, 1, 1) ys0 = y0[sel].view(m, 1, 1)