Spaces:
Running
Running
Adarsh7700
commited on
Commit
•
bbb6330
1
Parent(s):
73558ac
Update app.py
Browse files
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 ...')
|