Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Muennighoff
commited on
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,
|