Spaces:
Running
Running
Commit
·
17e48df
1
Parent(s):
4dd6c9a
Update app.py
Browse files
app.py
CHANGED
|
@@ -278,6 +278,7 @@ EXTERNAL_MODELS = [
|
|
| 278 |
"all-MiniLM-L6-v2",
|
| 279 |
"all-mpnet-base-v2",
|
| 280 |
"allenai-specter",
|
|
|
|
| 281 |
"bert-base-swedish-cased",
|
| 282 |
"bert-base-uncased",
|
| 283 |
"bge-base-zh-v1.5",
|
|
@@ -356,6 +357,7 @@ EXTERNAL_MODEL_TO_LINK = {
|
|
| 356 |
"all-MiniLM-L12-v2": "https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2",
|
| 357 |
"all-MiniLM-L6-v2": "https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2",
|
| 358 |
"all-mpnet-base-v2": "https://huggingface.co/sentence-transformers/all-mpnet-base-v2",
|
|
|
|
| 359 |
"bert-base-swedish-cased": "https://huggingface.co/KB/bert-base-swedish-cased",
|
| 360 |
"bert-base-uncased": "https://huggingface.co/bert-base-uncased",
|
| 361 |
"bge-base-zh-v1.5": "https://huggingface.co/BAAI/bge-base-zh-v1.5",
|
|
@@ -433,7 +435,8 @@ EXTERNAL_MODEL_TO_DIM = {
|
|
| 433 |
"all-MiniLM-L12-v2": 384,
|
| 434 |
"all-MiniLM-L6-v2": 384,
|
| 435 |
"all-mpnet-base-v2": 768,
|
| 436 |
-
"allenai-specter": 768,
|
|
|
|
| 437 |
"bert-base-swedish-cased": 768,
|
| 438 |
"bert-base-uncased": 768,
|
| 439 |
"bge-base-zh-v1.5": 768,
|
|
@@ -512,7 +515,8 @@ EXTERNAL_MODEL_TO_SEQLEN = {
|
|
| 512 |
"all-MiniLM-L6-v2": 512,
|
| 513 |
"all-mpnet-base-v2": 514,
|
| 514 |
"allenai-specter": 512,
|
| 515 |
-
"
|
|
|
|
| 516 |
"bert-base-uncased": 512,
|
| 517 |
"bge-base-zh-v1.5": 512,
|
| 518 |
"bge-large-zh-v1.5": 512,
|
|
|
|
| 278 |
"all-MiniLM-L6-v2",
|
| 279 |
"all-mpnet-base-v2",
|
| 280 |
"allenai-specter",
|
| 281 |
+
"Baichuan-text-embedding",
|
| 282 |
"bert-base-swedish-cased",
|
| 283 |
"bert-base-uncased",
|
| 284 |
"bge-base-zh-v1.5",
|
|
|
|
| 357 |
"all-MiniLM-L12-v2": "https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2",
|
| 358 |
"all-MiniLM-L6-v2": "https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2",
|
| 359 |
"all-mpnet-base-v2": "https://huggingface.co/sentence-transformers/all-mpnet-base-v2",
|
| 360 |
+
"Baichuan-text-embedding": "https://platform.baichuan-ai.com/docs/text-Embedding",
|
| 361 |
"bert-base-swedish-cased": "https://huggingface.co/KB/bert-base-swedish-cased",
|
| 362 |
"bert-base-uncased": "https://huggingface.co/bert-base-uncased",
|
| 363 |
"bge-base-zh-v1.5": "https://huggingface.co/BAAI/bge-base-zh-v1.5",
|
|
|
|
| 435 |
"all-MiniLM-L12-v2": 384,
|
| 436 |
"all-MiniLM-L6-v2": 384,
|
| 437 |
"all-mpnet-base-v2": 768,
|
| 438 |
+
"allenai-specter": 768,
|
| 439 |
+
"Baichuan-text-embedding": 1024,
|
| 440 |
"bert-base-swedish-cased": 768,
|
| 441 |
"bert-base-uncased": 768,
|
| 442 |
"bge-base-zh-v1.5": 768,
|
|
|
|
| 515 |
"all-MiniLM-L6-v2": 512,
|
| 516 |
"all-mpnet-base-v2": 514,
|
| 517 |
"allenai-specter": 512,
|
| 518 |
+
"Baichuan-text-embedding": 512,
|
| 519 |
+
"bert-base-swedish-cased": 512,
|
| 520 |
"bert-base-uncased": 512,
|
| 521 |
"bge-base-zh-v1.5": 512,
|
| 522 |
"bge-large-zh-v1.5": 512,
|