From f300c89bf5765fa3016dd83b4bf3bc02e44f3e5e Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Thu, 1 Aug 2024 21:26:02 +0800 Subject: [PATCH] add two official medical models --- README.md | 2 ++ docs/notes/pretrained_models.rst | 6 ++++++ 2 files changed, 8 insertions(+) diff --git a/README.md b/README.md index 74c8f51..8752863 100644 --- a/README.md +++ b/README.md @@ -76,6 +76,8 @@ BERT-based models: |----|------|------|------| | [WhereIsAI/UAE-Large-V1](https://huggingface.co/WhereIsAI/UAE-Large-V1) | 512 | cls | English, General-purpose | | [WhereIsAI/UAE-Code-Large-V1](https://huggingface.co/WhereIsAI/UAE-Code-Large-V1) | 512 | cls | Code Similarity | +| [WhereIsAI/pubmed-angle-base-en](https://huggingface.co/WhereIsAI/pubmed-angle-base-en) | 512 | cls | Medical Similarity | +| [WhereIsAI/pubmed-angle-base-en](https://huggingface.co/WhereIsAI/pubmed-angle-large-en) | 512 | cls | Medical Similarity | LLM-based models: diff --git a/docs/notes/pretrained_models.rst b/docs/notes/pretrained_models.rst index aa0ef4a..de3ccb9 100644 --- a/docs/notes/pretrained_models.rst +++ b/docs/notes/pretrained_models.rst @@ -13,9 +13,15 @@ BERT-based models: +------------------------------------+-------------+-------------------+--------------------------+ | `WhereIsAI/UAE-Code-Large-V1`_ | 512 | cls | Code Similarity | +------------------------------------+-------------+-------------------+--------------------------+ +| `WhereIsAI/pubmed-angle-base-en`_ | 512 | cls | Medical Similarity | ++------------------------------------+-------------+-------------------+--------------------------+ +| `WhereIsAI/pubmed-angle-large-en`_ | 512 | cls | Medical Similarity | ++------------------------------------+-------------+-------------------+--------------------------+ .. _WhereIsAI/UAE-Large-V1: https://huggingface.co/WhereIsAI/UAE-Large-V1 .. _WhereIsAI/UAE-Code-Large-V1: https://huggingface.co/WhereIsAI/UAE-Code-Large-V1 +.. _WhereIsAI/pubmed-angle-base-en: https://huggingface.co/WhereIsAI/pubmed-angle-base-en +.. _WhereIsAI/pubmed-angle-large-en: https://huggingface.co/WhereIsAI/pubmed-angle-large-en LLM-based models: