Muennighoff commited on
Commit
807cc67
1 Parent(s): abf58d4

Add voyage

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -448,6 +448,7 @@ EXTERNAL_MODELS = [
448
  "use-cmlm-multilingual",
449
  "voyage-2",
450
  "voyage-code-2",
 
451
  "voyage-law-2",
452
  "voyage-lite-01-instruct",
453
  "voyage-lite-02-instruct",
@@ -578,6 +579,7 @@ EXTERNAL_MODEL_TO_LINK = {
578
  "use-cmlm-multilingual": "https://huggingface.co/sentence-transformers/use-cmlm-multilingual",
579
  "voyage-2": "https://docs.voyageai.com/embeddings/",
580
  "voyage-code-2": "https://docs.voyageai.com/embeddings/",
 
581
  "voyage-law-2": "https://docs.voyageai.com/embeddings/",
582
  "voyage-lite-01-instruct": "https://docs.voyageai.com/embeddings/",
583
  "voyage-lite-02-instruct": "https://docs.voyageai.com/embeddings/",
@@ -707,6 +709,7 @@ EXTERNAL_MODEL_TO_DIM = {
707
  "use-cmlm-multilingual": 768,
708
  "voyage-2": 1024,
709
  "voyage-code-2": 1536,
 
710
  "voyage-law-2": 1024,
711
  "voyage-lite-01-instruct": 1024,
712
  "voyage-lite-02-instruct": 1024,
@@ -836,6 +839,7 @@ EXTERNAL_MODEL_TO_SEQLEN = {
836
  "unsup-simcse-bert-base-uncased": 512,
837
  "voyage-2": 1024,
838
  "voyage-code-2": 16000,
 
839
  "voyage-law-2": 4000,
840
  "voyage-lite-01-instruct": 4000,
841
  "voyage-lite-02-instruct": 4000,
@@ -1208,6 +1212,11 @@ MODELS_TO_SKIP = {
1208
  "McGill-NLP/LLM2Vec-Mistral-7B-Instruct-v2-mntp-unsup-simcse",
1209
  "McGill-NLP/LLM2Vec-Llama-2-7b-chat-hf-mntp-unsup-simcse",
1210
  "McGill-NLP/LLM2Vec-Sheared-LLaMA-mntp-unsup-simcse",
 
 
 
 
 
1211
  }
1212
 
1213
 
 
448
  "use-cmlm-multilingual",
449
  "voyage-2",
450
  "voyage-code-2",
451
+ "voyage-large-2-instruct",
452
  "voyage-law-2",
453
  "voyage-lite-01-instruct",
454
  "voyage-lite-02-instruct",
 
579
  "use-cmlm-multilingual": "https://huggingface.co/sentence-transformers/use-cmlm-multilingual",
580
  "voyage-2": "https://docs.voyageai.com/embeddings/",
581
  "voyage-code-2": "https://docs.voyageai.com/embeddings/",
582
+ "voyage-large-2-instruct": "https://docs.voyageai.com/embeddings/",
583
  "voyage-law-2": "https://docs.voyageai.com/embeddings/",
584
  "voyage-lite-01-instruct": "https://docs.voyageai.com/embeddings/",
585
  "voyage-lite-02-instruct": "https://docs.voyageai.com/embeddings/",
 
709
  "use-cmlm-multilingual": 768,
710
  "voyage-2": 1024,
711
  "voyage-code-2": 1536,
712
+ "voyage-large-2-instruct": 1536,
713
  "voyage-law-2": 1024,
714
  "voyage-lite-01-instruct": 1024,
715
  "voyage-lite-02-instruct": 1024,
 
839
  "unsup-simcse-bert-base-uncased": 512,
840
  "voyage-2": 1024,
841
  "voyage-code-2": 16000,
842
+ "voyage-large-2-instruct": 16000,
843
  "voyage-law-2": 4000,
844
  "voyage-lite-01-instruct": 4000,
845
  "voyage-lite-02-instruct": 4000,
 
1212
  "McGill-NLP/LLM2Vec-Mistral-7B-Instruct-v2-mntp-unsup-simcse",
1213
  "McGill-NLP/LLM2Vec-Llama-2-7b-chat-hf-mntp-unsup-simcse",
1214
  "McGill-NLP/LLM2Vec-Sheared-LLaMA-mntp-unsup-simcse",
1215
+ "jncraton/GIST-small-Embedding-v0-ct2-int8",
1216
+ "jncraton/stella-base-en-v2-ct2-int8",
1217
+ "lightbird-ai/nomic",
1218
+ "jamesdborin/jina-v2-base-en-embed",
1219
+ "iampanda/Test",
1220
  }
1221
 
1222