| 78 | |
| 79 | @dataclass |
| 80 | class FLAMEOutput(ModelOutput): |
| 81 | betas: Optional[Tensor] = None |
| 82 | expression: Optional[Tensor] = None |
| 83 | jaw_pose: Optional[Tensor] = None |
| 84 | neck_pose: Optional[Tensor] = None |
| 85 | |
| 86 | |
| 87 | def find_joint_kin_chain(joint_id, kinematic_tree): |