Spaces:
Running
on
Zero
Running
on
Zero
Update style_transfer.py
Browse files- style_transfer.py +1 -1
style_transfer.py
CHANGED
|
@@ -24,7 +24,7 @@ class DogStyleTransfer:
|
|
| 24 |
"""
|
| 25 |
def __init__(self):
|
| 26 |
self.models = {}
|
| 27 |
-
self.device = cpu
|
| 28 |
|
| 29 |
# Check xformers availability
|
| 30 |
self.xformers_available = False
|
|
|
|
| 24 |
"""
|
| 25 |
def __init__(self):
|
| 26 |
self.models = {}
|
| 27 |
+
self.device = 'cpu'
|
| 28 |
|
| 29 |
# Check xformers availability
|
| 30 |
self.xformers_available = False
|