yeraypabon

I'm also having this issue. My current workaround is to fall back to CPU using `import os` `os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'` at the beginning of your script. It does seems like...

I can confirm it works on Mac M3 too. Is there any way the code uses the parallel framework mac provides or it's only cuda?