Use MPS backend if available

isamplesorg / isamples_modelserver

REST API for invoking iSamples ML models

MIT License

0 stars 0 forks source link

Use MPS backend if available #3

Closed dannymandel closed 9 months ago

dannymandel commented 9 months ago

@hyunssong @datadavev -- this was the change I was experimenting with to use the MPS (GPU) backend on Apple Silicon macs. It's definitely faster than running on the CPU