BUG: fix embedding model gte-Qwen2 dimensions (#2479)

xorbitsai · Oct 24, 2024 · 6fd879c · 6fd879c
1 parent 0d10b38
commit 6fd879c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/doc/source/models/builtin/embedding/gte-qwen2.rst b/doc/source/models/builtin/embedding/gte-qwen2.rst
@@ -11,11 +11,11 @@ gte-Qwen2
 Specifications
 ^^^^^^^^^^^^^^
 
-- **Dimensions:** 3584
+- **Dimensions:** 4096
 - **Max Tokens:** 32000
 - **Model ID:** Alibaba-NLP/gte-Qwen2-7B-instruct
 - **Model Hubs**: `Hugging Face <https://huggingface.co/Alibaba-NLP/gte-Qwen2-7B-instruct>`__, `ModelScope <https://modelscope.cn/models/iic/gte_Qwen2-7B-instruct>`__
 
 Execute the following command to launch the model::
 
-   xinference launch --model-name gte-Qwen2 --model-type embedding
+   xinference launch --model-name gte-Qwen2 --model-type embedding
diff --git a/xinference/model/embedding/model_spec.json b/xinference/model/embedding/model_spec.json
@@ -233,7 +233,7 @@
   },
   {
     "model_name": "gte-Qwen2",
-    "dimensions": 3584,
+    "dimensions": 4096,
     "max_tokens": 32000,
     "language": ["zh", "en"],
     "model_id": "Alibaba-NLP/gte-Qwen2-7B-instruct",