File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 2424 fallback_device = torch .get_default_device ()
2525 torch .set_default_device ("mps" )
2626
27- # As of March 2025, the MacOS X-based GitHub Actions building enviroment sees
28- # the MPS GPU, but can't access it. Thus as try-except workaround is applied.
27+ # As of March 2025, the macOS X-based GitHub Actions building environment sees
28+ # the MPS GPU, but cannot access it. So, a try-except workaround is applied.
2929 try :
30- # A temporary trick to evade the Pytorch optimizer bug on MPS
30+ # A temporary trick to evade the Pytorch optimizer bug on MPS GPUs
3131 # See https://github.com/pytorch/pytorch/issues/149184
3232 torch ._dynamo .disable ()
3333
4646 import warnings
4747 warnings .warn (
4848 f'An MPS GPU has been detected, but cannot be used. '
49- f'Falling back to CPU.\n The exception message is:\n { e } '
49+ f'Falling back to the CPU.\n The exception message is:\n { e } '
5050 )
5151 torch .set_default_device (fallback_device )
5252
You can’t perform that action at this time.
0 commit comments