改回使用cuda

device = torch.device('cuda')

app.py

@@ -98,7 +98,7 @@ infer_config = config.infer_config
 
 IS_FLEXICUBES = True if config_name.startswith('instant-mesh') else False
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = torch.device('cuda')
 
 # load diffusion model
 print('Loading diffusion model ...')