Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
|
@@ -453,7 +453,7 @@ EXTERNAL_MODELS = [
|
|
| 453 |
"voyage-lite-02-instruct",
|
| 454 |
"xlm-roberta-base",
|
| 455 |
"xlm-roberta-large",
|
| 456 |
-
"
|
| 457 |
]
|
| 458 |
|
| 459 |
EXTERNAL_MODEL_TO_LINK = {
|
|
@@ -584,7 +584,7 @@ EXTERNAL_MODEL_TO_LINK = {
|
|
| 584 |
"voyage-lite-02-instruct": "https://docs.voyageai.com/embeddings/",
|
| 585 |
"xlm-roberta-base": "https://huggingface.co/xlm-roberta-base",
|
| 586 |
"xlm-roberta-large": "https://huggingface.co/xlm-roberta-large",
|
| 587 |
-
"
|
| 588 |
}
|
| 589 |
|
| 590 |
EXTERNAL_MODEL_TO_DIM = {
|
|
@@ -714,7 +714,7 @@ EXTERNAL_MODEL_TO_DIM = {
|
|
| 714 |
"voyage-lite-02-instruct": 1024,
|
| 715 |
"xlm-roberta-base": 768,
|
| 716 |
"xlm-roberta-large": 1024,
|
| 717 |
-
"
|
| 718 |
}
|
| 719 |
|
| 720 |
EXTERNAL_MODEL_TO_SEQLEN = {
|
|
@@ -844,7 +844,7 @@ EXTERNAL_MODEL_TO_SEQLEN = {
|
|
| 844 |
"voyage-lite-02-instruct": 4000,
|
| 845 |
"xlm-roberta-base": 514,
|
| 846 |
"xlm-roberta-large": 514,
|
| 847 |
-
"
|
| 848 |
}
|
| 849 |
|
| 850 |
EXTERNAL_MODEL_TO_SIZE = {
|
|
@@ -945,7 +945,7 @@ EXTERNAL_MODEL_TO_SIZE = {
|
|
| 945 |
"voyage-lite-02-instruct": 1220,
|
| 946 |
"xlm-roberta-base": 279,
|
| 947 |
"xlm-roberta-large": 560,
|
| 948 |
-
"
|
| 949 |
}
|
| 950 |
|
| 951 |
PROPRIETARY_MODELS = {
|
|
|
|
| 453 |
"voyage-lite-02-instruct",
|
| 454 |
"xlm-roberta-base",
|
| 455 |
"xlm-roberta-large",
|
| 456 |
+
"gte-Qwen1.5-7B-instruct"
|
| 457 |
]
|
| 458 |
|
| 459 |
EXTERNAL_MODEL_TO_LINK = {
|
|
|
|
| 584 |
"voyage-lite-02-instruct": "https://docs.voyageai.com/embeddings/",
|
| 585 |
"xlm-roberta-base": "https://huggingface.co/xlm-roberta-base",
|
| 586 |
"xlm-roberta-large": "https://huggingface.co/xlm-roberta-large",
|
| 587 |
+
"gte-Qwen1.5-7B-instruct": "https://huggingface.co/Alibaba-NLP/gte-Qwen1.5-7B-instruct"
|
| 588 |
}
|
| 589 |
|
| 590 |
EXTERNAL_MODEL_TO_DIM = {
|
|
|
|
| 714 |
"voyage-lite-02-instruct": 1024,
|
| 715 |
"xlm-roberta-base": 768,
|
| 716 |
"xlm-roberta-large": 1024,
|
| 717 |
+
"gte-Qwen1.5-7B-instruct": 4096
|
| 718 |
}
|
| 719 |
|
| 720 |
EXTERNAL_MODEL_TO_SEQLEN = {
|
|
|
|
| 844 |
"voyage-lite-02-instruct": 4000,
|
| 845 |
"xlm-roberta-base": 514,
|
| 846 |
"xlm-roberta-large": 514,
|
| 847 |
+
"gte-Qwen1.5-7B-instruct": 32768,
|
| 848 |
}
|
| 849 |
|
| 850 |
EXTERNAL_MODEL_TO_SIZE = {
|
|
|
|
| 945 |
"voyage-lite-02-instruct": 1220,
|
| 946 |
"xlm-roberta-base": 279,
|
| 947 |
"xlm-roberta-large": 560,
|
| 948 |
+
"gte-Qwen1.5-7B-instruct": 7099,
|
| 949 |
}
|
| 950 |
|
| 951 |
PROPRIETARY_MODELS = {
|