From 1fd5c58cff03b4257ada563795289300a49e9302 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sat, 22 Jun 2024 12:16:55 +0800 Subject: feat: ernie support embeddings and rereank (#630) --- models.yaml | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'models.yaml') diff --git a/models.yaml b/models.yaml index 87e1b14..867d0c0 100644 --- a/models.yaml +++ b/models.yaml @@ -515,6 +515,29 @@ require_max_tokens: true input_price: 0 output_price: 0 + - name: embedding-v1 + type: embedding + max_input_tokens: 384 + default_chunk_size: 700 + max_batch_size: 16 + - name: bge_large_zh + type: embedding + max_input_tokens: 512 + default_chunk_size: 1000 + max_batch_size: 16 + - name: bge_large_en + type: embedding + max_input_tokens: 512 + default_chunk_size: 1000 + max_batch_size: 16 + - name: tao_8k + type: embedding + max_input_tokens: 8192 + default_chunk_size: 2000 + max_batch_size: 1 + - name: bce_reranker_base + type: rerank + max_input_tokens: 1024 - platform: qianwen # docs: -- cgit v1.2.3