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.
This adds the cpu and cuda methods to make it easier to move the model to a different device. I also added a dtype property to make it easier to convert tensors to the same dtype as the model.
This adds the
cpu
andcuda
methods to make it easier to move the model to a different device. I also added adtype
property to make it easier to convert tensors to the same dtype as the model.This is preparation for fixing
to
.