Muennighoff commited on
Commit
e559e50
β€’
1 Parent(s): 3b7df4b

Add models

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -158,7 +158,9 @@ EXTERNAL_MODELS = [
158
  "sentence-t5-xxl",
159
  "sup-simcse-bert-base-uncased",
160
  "text-similarity-ada-001",
161
- "text-similarity-curie-001",
 
 
162
  "text-search-ada-001",
163
  "text-search-babbage-001",
164
  "text-search-curie-001",
@@ -168,7 +170,9 @@ EXTERNAL_MODELS = [
168
  EXTERNAL_MODEL_TO_LINK = {
169
  "LASER2": "https://github.com/facebookresearch/LASER",
170
  "text-similarity-ada-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
171
- "text-similarity-curie-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
 
 
172
  "text-search-ada-doc-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
173
  "text-search-ada-query-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
174
  "text-search-ada-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
@@ -227,7 +231,9 @@ EXTERNAL_MODEL_TO_DIM = {
227
  "sentence-t5-xxl": 768,
228
  "sup-simcse-bert-base-uncased": 768,
229
  "text-similarity-ada-001": 1024,
 
230
  "text-similarity-curie-001": 4096,
 
231
 
232
  "text-search-ada-doc-001": 1024,
233
  "text-search-ada-query-001": 1024,
 
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",
163
+ "text-similarity-davinci-001",
164
  "text-search-ada-001",
165
  "text-search-babbage-001",
166
  "text-search-curie-001",
 
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",
175
+ "text-similarity-davinci-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
176
  "text-search-ada-doc-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
177
  "text-search-ada-query-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
178
  "text-search-ada-001": "https://beta.openai.com/docs/guides/embeddings/types-of-embedding-models",
 
231
  "sentence-t5-xxl": 768,
232
  "sup-simcse-bert-base-uncased": 768,
233
  "text-similarity-ada-001": 1024,
234
+ "text-similarity-babbge-001": 2048,
235
  "text-similarity-curie-001": 4096,
236
+ "text-similarity-davinci-001": 12288,
237
 
238
  "text-search-ada-doc-001": 1024,
239
  "text-search-ada-query-001": 1024,