Spaces:

nbroad
/

bulk_embeddings

Paused

nbroad HF staff commited on Oct 25, 2023

Commit

92a399b

•

1 Parent(s): 09adc4d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,6 +12,7 @@ from utils import load_hf_dataset, get_model_and_tokenizer, batch_embed
 # "hkunlp/instructor-large",
 # "hkunlp/instructor-base",
 models_and_hidden_sizes = [
     ("intfloat/e5-small-v2", 384),
     ("intfloat/e5-base-v2", 768),
     ("intfloat/e5-large-v2", 1024),

 # "hkunlp/instructor-large",
 # "hkunlp/instructor-base",
 models_and_hidden_sizes = [
+    ("BAAI/bge-small-en-v1.5", 384),
     ("intfloat/e5-small-v2", 384),
     ("intfloat/e5-base-v2", 768),
     ("intfloat/e5-large-v2", 1024),