Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Vaibhav Adlakha
commited on
Commit
·
ae2f666
1
Parent(s):
c05fcec
adding metadata for llm2vec models
Browse files
app.py
CHANGED
|
@@ -341,6 +341,12 @@ EXTERNAL_MODELS = [
|
|
| 341 |
"DanskBERT",
|
| 342 |
"LASER2",
|
| 343 |
"LaBSE",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 344 |
"OpenSearch-text-hybrid",
|
| 345 |
"all-MiniLM-L12-v2",
|
| 346 |
"all-MiniLM-L6-v2",
|
|
@@ -453,6 +459,12 @@ EXTERNAL_MODEL_TO_LINK = {
|
|
| 453 |
"Cohere-embed-english-v3.0": "https://huggingface.co/Cohere/Cohere-embed-english-v3.0",
|
| 454 |
"Cohere-embed-multilingual-v3.0": "https://huggingface.co/Cohere/Cohere-embed-multilingual-v3.0",
|
| 455 |
"Cohere-embed-multilingual-light-v3.0": "https://huggingface.co/Cohere/Cohere-embed-multilingual-light-v3.0",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 456 |
"allenai-specter": "https://huggingface.co/sentence-transformers/allenai-specter",
|
| 457 |
"allenai-specter": "https://huggingface.co/sentence-transformers/allenai-specter",
|
| 458 |
"all-MiniLM-L12-v2": "https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2",
|
|
@@ -577,6 +589,12 @@ EXTERNAL_MODEL_TO_DIM = {
|
|
| 577 |
"Cohere-embed-english-v3.0": 1024,
|
| 578 |
"Cohere-embed-multilingual-v3.0": 1024,
|
| 579 |
"Cohere-embed-multilingual-light-v3.0": 384,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 580 |
"all-MiniLM-L12-v2": 384,
|
| 581 |
"all-MiniLM-L6-v2": 384,
|
| 582 |
"all-mpnet-base-v2": 768,
|
|
@@ -700,6 +718,12 @@ EXTERNAL_MODEL_TO_SEQLEN = {
|
|
| 700 |
"Cohere-embed-english-v3.0": 512,
|
| 701 |
"Cohere-embed-multilingual-v3.0": 512,
|
| 702 |
"Cohere-embed-multilingual-light-v3.0": 512,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 703 |
"all-MiniLM-L12-v2": 512,
|
| 704 |
"all-MiniLM-L6-v2": 512,
|
| 705 |
"all-mpnet-base-v2": 514,
|
|
@@ -820,6 +844,12 @@ EXTERNAL_MODEL_TO_SEQLEN = {
|
|
| 820 |
}
|
| 821 |
|
| 822 |
EXTERNAL_MODEL_TO_SIZE = {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 823 |
"allenai-specter": 110,
|
| 824 |
"all-MiniLM-L12-v2": 33,
|
| 825 |
"all-MiniLM-L6-v2": 23,
|
|
|
|
| 341 |
"DanskBERT",
|
| 342 |
"LASER2",
|
| 343 |
"LaBSE",
|
| 344 |
+
"LLM2Vec-Llama-supervised",
|
| 345 |
+
"LLM2Vec-Llama-unsupervised",
|
| 346 |
+
"LLM2Vec-Mistral-supervised",
|
| 347 |
+
"LLM2Vec-Mistral-unsupervised",
|
| 348 |
+
"LLM2Vec-Sheared-Llama-supervised",
|
| 349 |
+
"LLM2Vec-Sheared-Llama-unsupervised",
|
| 350 |
"OpenSearch-text-hybrid",
|
| 351 |
"all-MiniLM-L12-v2",
|
| 352 |
"all-MiniLM-L6-v2",
|
|
|
|
| 459 |
"Cohere-embed-english-v3.0": "https://huggingface.co/Cohere/Cohere-embed-english-v3.0",
|
| 460 |
"Cohere-embed-multilingual-v3.0": "https://huggingface.co/Cohere/Cohere-embed-multilingual-v3.0",
|
| 461 |
"Cohere-embed-multilingual-light-v3.0": "https://huggingface.co/Cohere/Cohere-embed-multilingual-light-v3.0",
|
| 462 |
+
"LLM2Vec-Llama-supervised": "https://huggingface.co/McGill-NLP/LLM2Vec-Llama-2-7b-chat-hf-mntp-supervised",
|
| 463 |
+
"LLM2Vec-Llama-unsupervised": "https://huggingface.co/McGill-NLP/LLM2Vec-Llama-2-7b-chat-hf-mntp",
|
| 464 |
+
"LLM2Vec-Mistral-supervised": "https://huggingface.co/McGill-NLP/LLM2Vec-Mistral-7B-Instruct-v2-mntp-supervised",
|
| 465 |
+
"LLM2Vec-Mistral-unsupervised": "https://huggingface.co/McGill-NLP/LLM2Vec-Mistral-7B-Instruct-v2-mntp",
|
| 466 |
+
"LLM2Vec-Sheared-Llama-supervised": "https://huggingface.co/McGill-NLP/LLM2Vec-Sheared-LLaMA-mntp-supervised",
|
| 467 |
+
"LLM2Vec-Sheared-Llama-unsupervised": "https://huggingface.co/McGill-NLP/LLM2Vec-Sheared-LLaMA-mntp",
|
| 468 |
"allenai-specter": "https://huggingface.co/sentence-transformers/allenai-specter",
|
| 469 |
"allenai-specter": "https://huggingface.co/sentence-transformers/allenai-specter",
|
| 470 |
"all-MiniLM-L12-v2": "https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2",
|
|
|
|
| 589 |
"Cohere-embed-english-v3.0": 1024,
|
| 590 |
"Cohere-embed-multilingual-v3.0": 1024,
|
| 591 |
"Cohere-embed-multilingual-light-v3.0": 384,
|
| 592 |
+
"LLM2Vec-Llama-supervised": 4096,
|
| 593 |
+
"LLM2Vec-Llama-unsupervised": 4096,
|
| 594 |
+
"LLM2Vec-Mistral-supervised": 4096,
|
| 595 |
+
"LLM2Vec-Mistral-unsupervised": 4096,
|
| 596 |
+
"LLM2Vec-Sheared-Llama-supervised": 2048,
|
| 597 |
+
"LLM2Vec-Sheared-Llama-unsupervised": 2048,
|
| 598 |
"all-MiniLM-L12-v2": 384,
|
| 599 |
"all-MiniLM-L6-v2": 384,
|
| 600 |
"all-mpnet-base-v2": 768,
|
|
|
|
| 718 |
"Cohere-embed-english-v3.0": 512,
|
| 719 |
"Cohere-embed-multilingual-v3.0": 512,
|
| 720 |
"Cohere-embed-multilingual-light-v3.0": 512,
|
| 721 |
+
"LLM2Vec-Llama-supervised": 512,
|
| 722 |
+
"LLM2Vec-Llama-unsupervised": 512,
|
| 723 |
+
"LLM2Vec-Mistral-supervised": 512,
|
| 724 |
+
"LLM2Vec-Mistral-unsupervised": 512,
|
| 725 |
+
"LLM2Vec-Sheared-Llama-supervised": 512,
|
| 726 |
+
"LLM2Vec-Sheared-Llama-unsupervised": 512,
|
| 727 |
"all-MiniLM-L12-v2": 512,
|
| 728 |
"all-MiniLM-L6-v2": 512,
|
| 729 |
"all-mpnet-base-v2": 514,
|
|
|
|
| 844 |
}
|
| 845 |
|
| 846 |
EXTERNAL_MODEL_TO_SIZE = {
|
| 847 |
+
"LLM2Vec-Llama-supervised": 6607,
|
| 848 |
+
"LLM2Vec-Llama-unsupervised": 6607,
|
| 849 |
+
"LLM2Vec-Mistral-supervised": 7111,
|
| 850 |
+
"LLM2Vec-Mistral-unsupervised": 7111,
|
| 851 |
+
"LLM2Vec-Sheared-Llama-supervised": 1280,
|
| 852 |
+
"LLM2Vec-Sheared-Llama-unsupervised": 1280,
|
| 853 |
"allenai-specter": 110,
|
| 854 |
"all-MiniLM-L12-v2": 33,
|
| 855 |
"all-MiniLM-L6-v2": 23,
|