Merge pull request vchoutas#146 from sha2nkt/patch-1

Fixes "TensorOutput has no 'v_shaped'"
suminmin · Jan 23, 2023 · 566532a · 566532a
2 parents bb595ae + 0d6f33c
commit 566532a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/smplx/body_models.py b/smplx/body_models.py
@@ -1136,7 +1136,7 @@ def forward(
         pose2rot: bool = True,
         return_shaped: bool = True,
         **kwargs
-    ) -> TensorOutput:
+    ) -> SMPLXOutput:
         '''
         Forward pass for the SMPLX model
 
@@ -1286,7 +1286,7 @@ def forward(
             v_shaped = self.v_template + blend_shapes(betas, self.shapedirs)
         else:
             v_shaped = Tensor(0)
-        output = TensorOutput(vertices=vertices if return_verts else None,
+        output = SMPLXOutput(vertices=vertices if return_verts else None,
                               joints=joints,
                               betas=betas,
                               expression=expression,