beamds.runs package#
Submodules#
beamds.runs.sentence_transformer_server module#
- class beamds.runs.sentence_transformer_server.SentenceTransformerServer(*args, _store_init_path=None, _save_init_args=True, **kwargs)[source]#
Bases:
BeamServeConfig- parameters = [BeamParam(name='tokenizer', type=<class 'str'>, default='BAAI/bge-base-en-v1.5', help='Tokenizer model', tags=None), BeamParam(name='st-model-path', type=<class 'str'>, default='BAAI/bge-base-en-v1.5', help='Dense model for text similarity', tags=None), BeamParam(name='st-model-device', type=<class 'str'>, default='cuda', help='Device for dense model', tags=None), BeamParam(name='st-kwargs', type=<class 'dict'>, default={}, help='additional kwargs for the sentence-transformer', tags=None)]#