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.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6cb669bed66b14cb4c11be51b89b28e17147b7027bb70091355466a9d4642e7c |
|
MD5 | 69e2ad5f099ba0612e187adea1565683 |
|
BLAKE2b-256 | d8451d5e20e3aac598bf8d6b5db54ed021ec8479aaff5ad42596061e271570a6 |