spandrel-extra-arches 0.1.0

Implements extra model architectures for spandrel

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

from spandrel import MAIN_REGISTRY, ModelLoader
from spandrel_extra_arches import EXTRA_REGISTRY

# add extra architectures before `ModelLoader` is used
MAIN_REGISTRY.add(*EXTRA_REGISTRY)

# load a model from disk
model = 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.