wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-26 01:39:25 +08:00
Code Issues Actions 31 Packages Projects Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
kijai 2026-06-09 09:37:56 +03:00
parent 2d514f5f0c
commit 57200a5e0f
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_extras/sam3d_body/utils.py
View File

@ -213,7 +213,7 @@ def run_batched_single_chunk(inner: SAM3DBody, frames_rgb: List[torch.Tensor], p
pose_output = outputs pose_output = outputs
batch_lhand = batch_rhand = None batch_lhand = batch_rhand = None
out = {k: v.cpu().numpy() for k, v in pose_output["mhr"].items() out = {k: v.float().cpu().numpy() for k, v in pose_output["mhr"].items()
if v is not None and k != "faces"} if v is not None and k != "faces"}
# Snapshot batch['bbox'] to CPU before we release `batch` references # Snapshot batch['bbox'] to CPU before we release `batch` references