Spaces:
Running
Running
Muennighoff
commited on
Commit
•
e559e50
1
Parent(s):
3b7df4b
Add models
Browse files
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-
|
|
|
|
|
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-
|
|
|
|
|
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,
|