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

Support APISR models #205

Closed RunDevelopment closed 6 months ago

RunDevelopment commented 6 months ago

Instead of the usual suspects, APISR uses "model_state_dict" as the key under which to store weights. While I don't think that we should support every possible key, "model_state_dict" is pretty much in line with the other keys, so I think it's fine to support this one.

Closes #199