Comment by regularfry

9 hours ago

It flat didn't work for me on mps. CUDA only until someone patches it.

Demo ran fine, if very slowly, with CPU-only using "--device cpu" for me. It defaults to CUDA though.

Try using mps I guess, I saw multiple references to code checking if device is not mps, so seems like it should be supported. If not, CPU.