Open Mr-Milk opened 2 months ago
See https://gist.github.com/priyathamkat/31d6b931095358840ba4bf073127864c for inspiration.
The idea is to simply launch the model on different devices. This makes it possible to run inference on both GPU and CPU.
See https://gist.github.com/priyathamkat/31d6b931095358840ba4bf073127864c for inspiration.
The idea is to simply launch the model on different devices. This makes it possible to run inference on both GPU and CPU.