Spaces:
Runtime error
Runtime error
Commit
·
98c1da8
1
Parent(s):
af428bc
update
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ if torch.cuda.is_available():
|
|
| 25 |
dtype = torch.float16
|
| 26 |
elif torch.backends.mps.is_available():
|
| 27 |
device = "mps"
|
| 28 |
-
|
| 29 |
else:
|
| 30 |
device = "cpu"
|
| 31 |
dtype = torch.float32
|
|
|
|
| 25 |
dtype = torch.float16
|
| 26 |
elif torch.backends.mps.is_available():
|
| 27 |
device = "mps"
|
| 28 |
+
dtype = torch.float16
|
| 29 |
else:
|
| 30 |
device = "cpu"
|
| 31 |
dtype = torch.float32
|