Designate a non-default cuda device to run inference?

[alfredcs]

By default AuraSR.from_pretrained() uses cuda:0. How to assign a diff device like cuda:1 for model inference?

[alfredcs]

Got it to work with setting CUDA_VISIBLE_DEVICES=1

[AliHaider20]

I'm trying to use it on MAC using MPS, but it doesn't work. Any solution for that?