Skip to main content

Implements extra model architectures for spandrel

Project description

Spandrel extra architectures

PyPI package version number PyPi Downloads Python Version

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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

spandrel_extra_arches-0.1.0-py3-none-any.whl (75.0 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page