Chaerin5 commited on
Commit
a6626fc
·
1 Parent(s): 063dd66

enable zerogpu

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -339,6 +339,8 @@ def get_ref_anno(ref):
339
  ]
340
  )
341
  image = image_transform(img)
 
 
342
  kpts_valid = check_keypoints_validity(keypts, opts.image_size)
343
  heatmaps = torch.tensor(
344
  keypoint_heatmap(
@@ -358,7 +360,7 @@ def get_ref_anno(ref):
358
  # device=device,
359
  ).unsqueeze(0)[None, ...]
360
  image, latent = make_ref_cond(
361
- image.numpy(),
362
  # keypts,
363
  # hand_mask,
364
  # device=device,
 
339
  ]
340
  )
341
  image = image_transform(img)
342
+ image = image.numpy()
343
+ image = torch.tensor(image)
344
  kpts_valid = check_keypoints_validity(keypts, opts.image_size)
345
  heatmaps = torch.tensor(
346
  keypoint_heatmap(
 
360
  # device=device,
361
  ).unsqueeze(0)[None, ...]
362
  image, latent = make_ref_cond(
363
+ image,
364
  # keypts,
365
  # hand_mask,
366
  # device=device,