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.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 08b3d2f24a91e8f2bcf765f6d1109ffd9545566ccac6f6ec189ac8952654805a |
|
MD5 | 8225bef39b229349802870296d68785e |
|
BLAKE2b-256 | 77cc6b50934db01901e438f392cd517cf8e084d7a40a108a30698dca475ca9b8 |