From e0b3c972d950c6633a5f5d3fdceafa2f9fd1f350 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 8 Apr 2022 14:19:13 +0800 Subject: [PATCH] Update readme Signed-off-by: Jael Gu --- README.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 93ed5f6..4e84be3 100644 --- a/README.md +++ b/README.md @@ -53,7 +53,13 @@ Create the operator via the following factory method The model name in string. The default value is "allenai/longformer-base-4096". -You can get the list of supported model names by calling `get_model_list` from [longformer.py](https://towhee.io/text-embedding/longformer/src/branch/main/longformer.py). + +Supported model names: +- allenai/longformer-base-4096 +- allenai/longformer-large-4096 +- allenai/longformer-large-4096-finetuned-triviaqa +- allenai/longformer-base-4096-extra.pos.embd.only +- allenai/longformer-large-4096-extra.pos.embd.only