Improve util error messages and fixed snapshot updates in certain cases

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

103 stars 7 forks source link

Improve util error messages and fixed snapshot updates in certain cases #220

Closed RunDevelopment closed 3 months ago

RunDevelopment commented 3 months ago

Changes:

Improve error messages for assert_size_requirements when the output tensor isn't the right shape.
Fixed snapshots not working when the expected and actual output don't have the same shape.