diff --git a/README.md b/README.md index a1310bb..173a673 100644 --- a/README.md +++ b/README.md @@ -352,6 +352,12 @@ 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 or list* + +​ The text embedding (or token embeddings) extracted by model. +
***save_model(format='pytorch', path='default')*** @@ -400,13 +406,6 @@ 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.