BentoML artifact framework for the Torchmoji Model
Project description
torchmojiartifact
BentoML artifact framework for the Torchmoji Model.
Installation:
pip install torchmojiartifact==0.0.1
Usage example (decorate service):
from torchmojiartifact.TorchmojiModelArtifact import TorchmojiModelArtifact
@artifacts([TorchmojiModelArtifact('emojify')])
class MyBentoService(BentoService):
Usage example (package model):
svc = MyBentoService()
svc.pack('emojify', model_path)
Alternatively, during training:
svc.pack('emojify', {'state_dict': my_state_dict, 'emoji_list': my_emoji_list, 'pax_list': my_pax_list, 'vocabulary': my_vocab})
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for torchmojiartifact-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c57740f49ff1bb4f954065d3a20a0ef8a39b275c3bc01af5d6ca901532a75d67 |
|
MD5 | 64532cf48ce447decf0615d0430c2f19 |
|
BLAKE2b-256 | 7678f9a8ccdc0af604eb932e37c807f58be4d1351f717cbe1168200a5c43b8c6 |