enable zerogpu

app.py

@@ -419,7 +419,7 @@ def get_target_anno(target):
             )
             * kpts_valid[:, None, None],
             dtype=torch.float,
-            device=device,
+            # device=device,
         )[None, ...]
         target_cond = torch.cat(
             [target_heatmaps, torch.zeros_like(target_heatmaps)[:, :1]], 1