Closed frankblubaugh closed 1 year ago
Adding this to _widget.py around line 32 should solve it. I can't figure out github's pull request nonsense:
if torch.cuda.is_available(): self._device = "cuda" elif torch.backends.mps.is_available(): self._device = "mps" else: self._device = "cpu"
Hi @frankblubaugh , thanks for the contribution. Can you check if this branch works for you? Once I get some feedback that it works on "mps" I'll merge it into main.
My colleague tested it, closed #9
add mac gpu support if available