From 04cc8fa80ec53907a985f26ea39d1e028a5c973a Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 20 Apr 2025 12:53:30 +0300 Subject: [PATCH] Update nodes.py --- unianimate/nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unianimate/nodes.py b/unianimate/nodes.py index 2b873ca..536a09c 100644 --- a/unianimate/nodes.py +++ b/unianimate/nodes.py @@ -218,7 +218,7 @@ def pose_extract(pose_images, ref_image, dwpose_model, height, width, score_thre try: pose = dwpose_model(img, score_threshold=score_threshold) except: - pose = torch.zeros_like(img) + pose = np.zeros_like(img) results_vis.append(pose) comfy_pbar.update(1)