Update wilor/models/__init__.py

wilor/models/__init__.py

@@ -32,5 +32,5 @@ def load_wilor(checkpoint_path, cfg_path):
         model_cfg.MANO.MEAN_PARAMS = './mano_data/mano_mean_params.npz'
         model_cfg.freeze()
 
-    model = WiLoR.load_from_checkpoint(checkpoint_path, strict=False, cfg=model_cfg)
+    model = WiLoR.load_from_checkpoint(checkpoint_path, strict=False, cfg=model_cfg, map_location=torch.device("cpu"))
     return model, model_cfg