Spaces:

alexnasa
/

pixel3dmm

Running on Zero

alexnasa commited on Jun 11

Commit

2b8469f

verified ·

1 Parent(s): 8355ef8

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

Files changed (1) hide show

src/pixel3dmm/tracking/flame/lbs.py CHANGED Viewed

@@ -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
     # Add shape contribution
     v_shaped = v_template + blend_shapes(betas, shapedirs)

     batch_size = max(betas.shape[0], pose.shape[0])
+    device = betas.get_device()
     # Add shape contribution
     v_shaped = v_template + blend_shapes(betas, shapedirs)