Browse Source
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
2 changed files with
3 additions and
3 deletions
-
README.md
-
__init__.py
|
|
@ -13,7 +13,7 @@ A text embedding operator implemented with pretrained models from [Huggingface T |
|
|
```python |
|
|
```python |
|
|
from towhee import ops |
|
|
from towhee import ops |
|
|
|
|
|
|
|
|
text_encoder = ops.text_embedding.transformers("bert-base-cased") |
|
|
|
|
|
|
|
|
text_encoder = ops.text_embedding.transformers(model_name="bert-base-cased") |
|
|
text_embedding = text_encoder("Hello, world.") |
|
|
text_embedding = text_encoder("Hello, world.") |
|
|
``` |
|
|
``` |
|
|
|
|
|
|
|
|
|
|
|
@ -15,5 +15,5 @@ |
|
|
from .auto_transformers import AutoTransformers |
|
|
from .auto_transformers import AutoTransformers |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def transformers(): |
|
|
|
|
|
return AutoTransformers() |
|
|
|
|
|
|
|
|
def transformers(model_name: str): |
|
|
|
|
|
return AutoTransformers(model_name) |
|
|
|
