| 1208 | |
| 1209 | class SMPLXLayer(SMPLX): |
| 1210 | def __init__(self, *args, **kwargs) -> None: |
| 1211 | # Just create a SMPLX module without any member variables |
| 1212 | super(SMPLXLayer, self).__init__( |
| 1213 | create_global_orient=False, |
| 1214 | create_body_pose=False, |
| 1215 | create_left_hand_pose=False, |
| 1216 | create_right_hand_pose=False, |
| 1217 | create_jaw_pose=False, |
| 1218 | create_leye_pose=False, |
| 1219 | create_reye_pose=False, |
| 1220 | create_betas=False, |
| 1221 | create_expression=False, |
| 1222 | create_transl=False, |
| 1223 | *args, |
| 1224 | **kwargs, |
| 1225 | ) |
| 1226 | |
| 1227 | def forward(self, |
| 1228 | betas: Optional[Tensor] = None, |