diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-22 12:16:55 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-22 12:16:55 +0800 |
| commit | 1fd5c58cff03b4257ada563795289300a49e9302 (patch) | |
| tree | 3f1ae848657bfa9ee26a405f4dc75d6f1312ec0f /models.yaml | |
| parent | de16813beeffe2474572e51e851f012bc23e1d4f (diff) | |
| download | aichat-1fd5c58cff03b4257ada563795289300a49e9302.tar.gz | |
feat: ernie support embeddings and rereank (#630)
Diffstat (limited to 'models.yaml')
| -rw-r--r-- | models.yaml | 23 |
1 files changed, 23 insertions, 0 deletions
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: |