BentoML artifact framework for Q&A Transformers
Project description
qandaxfmrartifact
BentoML artifact framework for Q&A Transformers.
Installation:
pip install qandaxfmrartifact==0.0.1
Usage example (decorate service):
from qandaxfmrartifact.QandaTransformersModelArtifact import QandaTransformersModelArtifact
@artifacts([QandaTransformersModelArtifact('albert')])
class MyBentoService(BentoService):
Usage example (package model):
svc = MyBentoService()
opts = {
'embedder_model_path': my_embedder_model_path,
}
svc.pack('albert', my_transformer_model_path, opts)
Alternatively, during training:
svc.pack('albert', {'model': my_trained_model, 'tokenizer': my_tokenizer, 'embedder': my_embedder})
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
Built Distribution
Close
Hashes for qandaxfmrartifact-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 55cdb39a7762bfdd86d5ce90b0d129c28a04f315e2d9d1e22fc6e22a0d83f456 |
|
MD5 | ba074e86e6b226ca6012684ce3e69f14 |
|
BLAKE2b-256 | 9eaed91d715a5a9db89e007c261bbdaca3c123f8c4a9ab0f153acdf3ccba11e0 |