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.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 33a6c395ba9fd7ffd847fee1029c5c3307bfabe4dcb30f2709a7d7a2b281fdce |
|
MD5 | e88e3e7f8cfc0345fcd916935ceb7396 |
|
BLAKE2b-256 | eb37c3d29a43898cfc6b5f20714fdc5e3c31d3ee3a1f63dbfedf4896f455f3eb |