Adding HF files

hmr2/utils/utils_detectron2.py

@@ -67,7 +67,8 @@ class DefaultPredictor_Lazy:
             self.aug = mapper.augmentations
             self.input_format = mapper.image_format
 
-        self.model.eval().cuda()
+        self.device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
+        self.model.eval().to(self.device)
         if test_dataset:
             self.metadata = MetadataCatalog.get(test_dataset)
         assert self.input_format in ["RGB", "BGR"], self.input_format

requirements.txt

@@ -9,9 +9,9 @@
 # timm
 # OmegaConf
 
---extra-index-url https://download.pytorch.org/whl/cu116
-torch==1.13.1+cu116
-torchvision==0.14.1+cu116
+# --extra-index-url https://download.pytorch.org/whl/cu116
+torch==1.13.1#+cu116
+torchvision==0.14.1#+cu116
 pytorch-lightning
 smplx==0.1.28
 opencv-python