Allow strings for Torch devices, like Torch does

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

Allow strings for Torch devices, like Torch does #112

Closed akx closed 9 months ago

akx commented 9 months ago

As discussed in https://github.com/chaiNNer-org/spandrel/pull/111#issuecomment-1874572986.

Torch's to also allows integers but I think that's taking things too far.