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.
So I changed our API to allow str devices again. The if isinstance(device, str): device = torch.device(device) is only necessary to make the type checker happy.
PyTorch docs and types are wrong. See https://github.com/pytorch/pytorch/issues/117936. So #133 was misguided.
So I changed our API to allow str devices again. The
if isinstance(device, str): device = torch.device(device)
is only necessary to make the type checker happy.