Spaces:
Running
Running
xco2
commited on
Commit
·
c76f1ef
1
Parent(s):
f802d8b
改成cpu运行
Browse files
app.py
CHANGED
|
@@ -531,7 +531,7 @@ def init_webui(unet, vae, normal_t):
|
|
| 531 |
|
| 532 |
|
| 533 |
if __name__ == '__main__':
|
| 534 |
-
device = "
|
| 535 |
unet, vae, normal_t = get_models(device)
|
| 536 |
|
| 537 |
|
|
|
|
| 531 |
|
| 532 |
|
| 533 |
if __name__ == '__main__':
|
| 534 |
+
device = "cpu"
|
| 535 |
unet, vae, normal_t = get_models(device)
|
| 536 |
|
| 537 |
|