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

Add support for MPRNet #210

Closed RunDevelopment closed 6 months ago

RunDevelopment commented 6 months ago

This PR adds support for MPRNet, an older arch by the same authors as Restormer. All hyperparameters are detected. Interestingly, this arch does not support safetensors.