search

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
112 stars 7 forks source link

tests: add device and outputs directory environment variables #111

Closed akx closed 6 months ago

akx commented 6 months ago

Please see within.

joeyballentine commented 6 months ago

Ah, wrap the device name by doing torch.device(device_name) and that should fix the type error

akx commented 6 months ago

@joeyballentine Sure – but factually, .to() also works with a device name, so should the signature be adjusted to allow for that?

joeyballentine commented 6 months ago

Yeah that should probably be adjusted to match whatever torch's signature is