Switch our model loading code to use Spandrel + support FBCNN (#2330)

* initial spandrel implementation * bump spandrel to 0.0.2 * fix pytorch inputs * idk how this happened * fix some type issues * Add FBCNN support to upscale image * add missing future annotations * switch to using union * Apply suggestions from code review Co-authored-by: Michael Schmidt <mitchi5000.ms@googlemail.com> * PR suggestions * future proof saving function (assuming we get rid of the state_dict property * update spandrel --------- Co-authored-by: Michael Schmidt <mitchi5000.ms@googlemail.com>
chaiNNer-org · Nov 21, 2023 · 9850f9c · 9850f9c
1 parent e4cee69
commit 9850f9c
Show file tree

Hide file tree

Showing 70 changed files with 162 additions and 22,017 deletions.
diff --git a/backend/src/nodes/impl/pytorch/architecture/DAT.py b/backend/src/nodes/impl/pytorch/architecture/DAT.py
diff --git a/backend/src/nodes/impl/pytorch/architecture/HAT.py b/backend/src/nodes/impl/pytorch/architecture/HAT.py
diff --git a/backend/src/nodes/impl/pytorch/architecture/LICENSE-DAT b/backend/src/nodes/impl/pytorch/architecture/LICENSE-DAT