Muennighoff commited on
Commit
a90dc55
1 Parent(s): b7b70f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -157,6 +157,7 @@ EXTERNAL_MODELS = [
157
  "sentence-t5-xl",
158
  "sentence-t5-xxl",
159
  "sup-simcse-bert-base-uncased",
 
160
  "text-similarity-ada-001",
161
  "text-similarity-babbage-001",
162
  "text-similarity-curie-001",
@@ -169,6 +170,7 @@ EXTERNAL_MODELS = [
169
  ]
170
  EXTERNAL_MODEL_TO_LINK = {
171
  "LASER2": "https://github.com/facebookresearch/LASER",
 
172
  "text-similarity-ada-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
173
  "text-similarity-babbage-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
174
  "text-similarity-curie-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
@@ -230,6 +232,9 @@ EXTERNAL_MODEL_TO_DIM = {
230
  "sentence-t5-xl": 768,
231
  "sentence-t5-xxl": 768,
232
  "sup-simcse-bert-base-uncased": 768,
 
 
 
233
  "text-similarity-ada-001": 1024,
234
  "text-similarity-babbage-001": 2048,
235
  "text-similarity-curie-001": 4096,
 
157
  "sentence-t5-xl",
158
  "sentence-t5-xxl",
159
  "sup-simcse-bert-base-uncased",
160
+ "text-embedding-ada-002",
161
  "text-similarity-ada-001",
162
  "text-similarity-babbage-001",
163
  "text-similarity-curie-001",
 
170
  ]
171
  EXTERNAL_MODEL_TO_LINK = {
172
  "LASER2": "https://github.com/facebookresearch/LASER",
173
+ "text-embedding-ada-002": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
174
  "text-similarity-ada-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
175
  "text-similarity-babbage-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
176
  "text-similarity-curie-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
 
232
  "sentence-t5-xl": 768,
233
  "sentence-t5-xxl": 768,
234
  "sup-simcse-bert-base-uncased": 768,
235
+
236
+ "text-embedding-ada-002": 1536,
237
+
238
  "text-similarity-ada-001": 1024,
239
  "text-similarity-babbage-001": 2048,
240
  "text-similarity-curie-001": 4096,