remove map loc

app.py

@@ -128,7 +128,7 @@ args = parser.parse_args()
 crm_path = hf_hub_download(repo_id="Zhengyi/CRM", filename="CRM.pth")
 specs = json.load(open("configs/specs_objaverse_total.json"))
 model = CRM(specs).to(args.device)
-model.load_state_dict(torch.load(crm_path, map_location = args.device), strict=False)
+model.load_state_dict(torch.load(crm_path), strict=False)
 
 stage1_config = OmegaConf.load(args.stage1_config).config
 stage2_config = OmegaConf.load(args.stage2_config).config