From 00e0c3bd4fcc956b1956b934ab42ae0fca8321fc Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Tue, 13 Dec 2022 16:31:42 +0800 Subject: [PATCH] Update README Signed-off-by: Jael Gu --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index d37c853..830d27a 100644 --- a/README.md +++ b/README.md @@ -331,6 +331,7 @@ and then return text embedding in ndarray. ​ The text embedding extracted by model. +
***save_model(format='pytorch', path='default')*** @@ -342,10 +343,11 @@ Save model to local with specified format. ​ The format of saved model, defaults to 'pytorch'. -***format***: *path* +***path***: *str* ​ The path where model is saved to. By default, it will save model to the operator directory. +
***supported_model_names(format=None)***