Update src/pixel3dmm/tracking/flame/lbs.py

src/pixel3dmm/tracking/flame/lbs.py

@@ -209,8 +209,7 @@ def lbs(betas, pose, v_template, shapedirs, posedirs, J_regressor, parents,
 
     batch_size = max(betas.shape[0], pose.shape[0])
 
-    # after: get the device index (an int), which Dynamo can capture
-    device = betas.device
+    device = betas.get_device()
     
     # Add shape contribution
     v_shaped = v_template + blend_shapes(betas, shapedirs)