Spaces:
Runtime error
Runtime error
File size: 370 Bytes
0140c70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import torch
def detect_device():
"""
Detects the appropriate device to run on, and return the device and dtype.
"""
if torch.cuda.is_available():
return torch.device("cuda"), torch.float16
elif torch.backends.mps.is_available():
return torch.device("mps"), torch.float16
else:
return torch.device("cpu"), torch.float32
|