From 0635dd44194a7962b2b0d9af47c96f5c5dca0e14 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 6 Jan 2023 14:45:52 +0800 Subject: [PATCH] Update README Signed-off-by: Jael Gu --- README.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/README.md b/README.md index af572db..3ce47ab 100644 --- a/README.md +++ b/README.md @@ -350,6 +350,10 @@ and then return text embedding in ndarray. ​ The text embedding extracted by model. +***return_sentence_emb***: *bool* + +The flag to output sentence embedding instead of token embeddings, defaults to False. +
***save_model(format='pytorch', path='default')***