Commit
·
9da9b60
1
Parent(s):
2effeef
update
Browse files
app.py
CHANGED
|
@@ -56,8 +56,6 @@ HEADER = """
|
|
| 56 |
|
| 57 |
if torch.cuda.is_available():
|
| 58 |
device = "cuda:0"
|
| 59 |
-
elif torch.backends.mps.is_available():
|
| 60 |
-
device = "mps"
|
| 61 |
else:
|
| 62 |
device = "cpu"
|
| 63 |
|
|
|
|
| 56 |
|
| 57 |
if torch.cuda.is_available():
|
| 58 |
device = "cuda:0"
|
|
|
|
|
|
|
| 59 |
else:
|
| 60 |
device = "cpu"
|
| 61 |
|