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