Add `device` property for models

chaiNNer-org / spandrel

Spandrel gives your project support for various PyTorch architectures meant for AI Super-Resolution, restoration, and inpainting. Based on the model support implemented in chaiNNer.

MIT License

139 stars 12 forks source link

Add `device` property for models #114

Closed RunDevelopment closed 9 months ago

RunDevelopment commented 9 months ago

This adds a device property to conveniently get the device of a model.