FuriosaAI model registry
Project description
Furiosa Registry
Furiosa Registry defines Model
which allows client to communicate with several types of registries.
Getting started
Install
pip install furiosa-registry
List up available models
import furiosa.common.thread as synchronous
import furiosa.registry as registry
# Repository where to load models from.
repository = "https://github.com/furiosa-ai/furiosa-artifacts"
# List up the available artifacts.
models: List[registry.Model] = synchronous(registry.list)(repository)
for model in models:
print(model.name)
Load models
import furiosa.common.thread as synchronous
import furiosa.registry as registry
# Repository where to load models from.
repository = "https://github.com/furiosa-ai/furiosa-artifacts"
models: registry.Model = synchronous(registry.load)(uri=repository, name="MLCommonsResNet50")
# Access the model
print(model.name)
print(model.version)
print(model.metadata.description)
Print documentation about a model
import furiosa.common.thread as synchronous
import furiosa.registry as registry
# Repository where to load models from.
repository = "https://github.com/furiosa-ai/furiosa-artifacts"
print(synchronous(registry.help)(repository, "MLCommonsResNet50"))
Development
Generate artfiact JSON schema from pydantic model definition.
When you changed model schema, you can generate modified schema via
python -c 'from furiosa.registry import Model;\
print(Model.schema_json(indent=2), file=open("model_schema.json", "w"))'
See model_schema.json
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
furiosa-registry-0.9.0rc2.tar.gz
(70.9 kB
view hashes)
Built Distribution
Close
Hashes for furiosa-registry-0.9.0rc2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 520523aa5fc9e27e3faaff3e94a6d34a73d7dcf11ec1c08d4e0c8a941964e319 |
|
MD5 | 1b18df31ff9d38c66f29f0d42c9c3ec0 |
|
BLAKE2b-256 | 5171aa61e53154cdea016c62ed7de99e6e1da91da644c2b2ece4599a95bc2954 |
Close
Hashes for furiosa_registry-0.9.0rc2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 05aa3a49a1e3d038db15f33c6f902dcf2ac57937f6b295b74a0507c66a682b61 |
|
MD5 | 40607b1b88fae030988875f6732a1bf3 |
|
BLAKE2b-256 | 517ce962cb452c079f046e99a6f896c0ff3d81fa300b0cc8fdf5fa361a2f8865 |