Adarsh7700 commited on
Commit
bbb6330
1 Parent(s): 73558ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -98,7 +98,8 @@ infer_config = config.infer_config
98
 
99
  IS_FLEXICUBES = True if config_name.startswith('instant-mesh') else False
100
 
101
- device = torch.device('cuda')
 
102
 
103
  # load diffusion model
104
  print('Loading diffusion model ...')
 
98
 
99
  IS_FLEXICUBES = True if config_name.startswith('instant-mesh') else False
100
 
101
+ # device = torch.device('cuda')
102
+ device = torch.device('cpu')
103
 
104
  # load diffusion model
105
  print('Loading diffusion model ...')