From f40051c271156b38458d295a18ea1d60350e3539 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?H=2ED=2ET=C3=A0i?= Date: Fri, 26 Apr 2024 12:47:15 +0700 Subject: [PATCH] Update spectre_nodes.py --- nodes/spectre_nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes/spectre_nodes.py b/nodes/spectre_nodes.py index aea4ae7..f5efebc 100644 --- a/nodes/spectre_nodes.py +++ b/nodes/spectre_nodes.py @@ -265,10 +265,10 @@ class SpectreImg2SMPL: return ((all_verts, { "frame_width": 224, "frame_height": 224, "vertical_flip": True, "focal_length": focal_length, - "faces": spectre.flame.faces_tensor + "faces": spectre.flame.faces_tensor.cpu().detach() }), all_cropped_images) NODE_CLASS_MAPPINGS = { "SpectreFaceReconLoader": SpectreFaceReconLoader, "SpectreImg2SMPL": SpectreImg2SMPL -} \ No newline at end of file +}