Muennighoff commited on
Commit
17e48df
1 Parent(s): 4dd6c9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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
- "bert-base-swedish-cased": 512,
 
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,