spandrel-extra-arches

Spandrel extra architectures

This library implements various PyTorch architectures with restrictive licenses for spandrel.

If you are working on a private project or non-commercial open-source project, you are free to use this library. If you are working on a commercial or closed-source project, you may need to review the licenses of the individual architectures before using this library.

Installation

You need to install both spandrel and spandrel_extra_arches:

pip install spandrel spandrel_extra_arches

Basic usage

import spandrel
import spandrel_extra_arches

# add extra architectures before `ModelLoader` is used
spandrel_extra_arches.install()

# load a model from disk
model = spandrel.ModelLoader().load_from_file(r"path/to/model.pth")

... # use model

Model Architecture Support

For a full list of all architectures implemented in this library, see the architectures marked with "(+)" here.

License

This library is licensed under the MIT license but contains the source code of architectures with non-commercial and copyleft licenses. You may need to review the licenses of the individual architectures before using this library.