From 57200a5e0f0f442b65dbb6b300ad0617551f00ba Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 9 Jun 2026 09:37:56 +0300 Subject: [PATCH] Update utils.py --- comfy_extras/sam3d_body/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy_extras/sam3d_body/utils.py b/comfy_extras/sam3d_body/utils.py index 576f0d340..5fff80953 100644 --- a/comfy_extras/sam3d_body/utils.py +++ b/comfy_extras/sam3d_body/utils.py @@ -213,7 +213,7 @@ def run_batched_single_chunk(inner: SAM3DBody, frames_rgb: List[torch.Tensor], p pose_output = outputs 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"} # Snapshot batch['bbox'] to CPU before we release `batch` references