From 3085df8bafec4edb9df8b447d4337e660bd75986 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 6 Jan 2023 17:00:35 +0800 Subject: [PATCH] Update readme Signed-off-by: Jael Gu --- README.md | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index 3bbdbb4..a1310bb 100644 --- a/README.md +++ b/README.md @@ -352,14 +352,6 @@ and then return text embedding in ndarray. If data is string, the operator returns embedding(s) in ndarray. If data is a list, the operator returns embedding(s) in a list. - -**Returns**: - -*numpy.ndarray* - -​ The text embedding extracted by model. - -
***save_model(format='pytorch', path='default')*** @@ -408,6 +400,16 @@ print(f'Onnx-support/Total Models: {len(onnx_list)}/{len(full_list)}') 2022-12-13 16:25:15,916 - 140704500614336 - auto_transformers.py-auto_transformers:68 - WARNING: The operator is initialized without specified model. Onnx-support/Total Models: 111/126 +
+ +**Returns**: + +*numpy.ndarray or list* + +​ The text embedding (or token embeddings) extracted by model. + +
+ ## Fine-tune ### Requirement If you want to train this operator, besides dependency in requirements.txt, you need install these dependencies.