FIx device select and add mps device

CNChTu / FCPE

MIT License

83 stars 16 forks source link

Closed Tps-F closed 6 months ago

Tps-F commented 6 months ago

What to do when mps or cuda device does not work.
added mps to auto select (I used PYTORCH_ENABLE_MPS_FALLBACK=1 and it worked).

CNChTu commented 6 months ago

thank you for your pr I marged it