Change default device setting to CPU in app.py for improved compatibility. Updated model loading to ensure it operates on CPU, enhancing performance on systems without GPU support.

app.py

@@ -154,14 +154,15 @@ parser.add_argument(
     help="config for stage2",
 )
 
-parser.add_argument("--device", type=str, default="cuda")
+# Force CPU usage
+parser.add_argument("--device", type=str, default="cpu")
 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)
 model.load_state_dict(torch.load(crm_path, map_location="cpu"), strict=False)
-model = model.to(args.device)
+model = model.to("cpu")
 
 stage1_config = OmegaConf.load(args.stage1_config).config
 stage2_config = OmegaConf.load(args.stage2_config).config