change map location to cpu

pq3d/inference.py

@@ -169,7 +169,7 @@ def inference(scan_id, text):
     data_dict.update(tokenize_txt(text))
     data_dict = form_batch(data_dict)
     model = Query3DUnified()
-    load_msg = model.load_state_dict(torch.load(os.path.join(CKPT_DIR, 'pytorch_model.bin')), strict=False)
+    load_msg = model.load_state_dict(torch.load(os.path.join(CKPT_DIR, 'pytorch_model.bin'), map_location='cpu'), strict=False)
     data_dict = model(data_dict)
     result_id = data_dict['obj_ids'][0][torch.argmax(data_dict['og3d_logits'][0]).item()]
     print(f"finish infernece result id is {result_id}")