how to use cpu/mps?

[LukaGiorgadze]

Hey there! I'm on Apple M2 Mac.
How can I use mps/cpu device only?

When I'm running example, getting this error:

raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

[Algolzw]

It might be hard to run this model on Mac since it requires a lot of memory and would be very slow on the CPU.

[LukaGiorgadze]

It might be hard to run this model on Mac since it requires a lot of memory and would be very slow on the CPU.

Probably worth trying. I'm already running image matting models without any issues.
Well, basically we just need to pass --device param (gpu/cpu/mps) instead of hard-coding cuda.

I would love to see this in the next commits.

Thoughts?

[Algolzw]

Good to hear your models are running! Thanks, then I will consider adding the "device" param in the future.