From c8fdfd6dab45d16710498eeb6d1cd3e893ac611a Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 30 Jul 2024 22:51:45 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nodes.py b/nodes.py index e340cb4..4100200 100644 --- a/nodes.py +++ b/nodes.py @@ -304,6 +304,8 @@ class Florence2Run: annotated_image_tensor = F.to_tensor(annotated_image_pil) out_tensor = annotated_image_tensor[:3, :, :].unsqueeze(0).permute(0, 2, 3, 1).cpu().float() out.append(out_tensor) + + out_data.append(bboxes) pbar.update(1)