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 +}