Spaces:
Build error
Build error
Commit
·
302fb80
1
Parent(s):
a534a2d
Add files
Browse files
app.py
CHANGED
|
@@ -256,7 +256,8 @@ class wrapper(object):
|
|
| 256 |
tag_lora=None,
|
| 257 |
tag_scheduler=None,):
|
| 258 |
|
| 259 |
-
self.device = "cuda"
|
|
|
|
| 260 |
if fp16:
|
| 261 |
self.torch_dtype = torch.float16
|
| 262 |
else:
|
|
|
|
| 256 |
tag_lora=None,
|
| 257 |
tag_scheduler=None,):
|
| 258 |
|
| 259 |
+
# self.device = "cuda"
|
| 260 |
+
self.device = "cpu"
|
| 261 |
if fp16:
|
| 262 |
self.torch_dtype = torch.float16
|
| 263 |
else:
|