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.
A lot of the .pt files I downloaded were actually .pth files. Since this seems to be a common problem, I added support for this in ModelLoader#load_state_dict_from_file. When we fail to load a .pt file, we will now try to load it as a .pth file.
Other changes:
Ever since #40, the implementation of _load_ckpt and _load_pth is the same, so I removed _load_ckpt.
A lot of the
.ptfiles I downloaded were actually.pthfiles. Since this seems to be a common problem, I added support for this inModelLoader#load_state_dict_from_file. When we fail to load a.ptfile, we will now try to load it as a.pthfile.Other changes:
_load_ckptand_load_pthis the same, so I removed_load_ckpt.