From a20b4b1d5010c69172afad630b1909eea94e4650 Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Fri, 5 May 2023 14:00:24 +0000
Subject: [PATCH] fix smpl when running on garments

---
 smpl/smpl_torch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/smpl/smpl_torch.py b/smpl/smpl_torch.py
index f2a87cb..16e1855 100644
--- a/smpl/smpl_torch.py
+++ b/smpl/smpl_torch.py
@@ -218,7 +218,7 @@ class SMPL(Module):
     result = result - root_joint
     joints = joints - root_joint
 
-    if remove:
+    if remove and verts is None:
       result = result[:,self.to_keep,:]
 
     return result, joints