update readme

Author: shell-nlp

README.md

@@ -293,26 +293,27 @@ Chat UI界面:
 
 以下模型经过测试可放心使用：
 
-| Models / BackEnd                                                           | sentence_transformers  | infinity | vllm|
-| ---------------------------------------------------- | --- | -------- |-------- |
-| bge-m3                                                                        | √   | √        |×        |
-| bge-vl                                                                   | √   | ×        |×        |
-| jina-reranker-m0                                                                   | √   | ×        |×        |
+| Models / BackEnd                                                                    | sentence_transformers  | infinity | vllm|
+| ----------------------------------------------------------------------------------- | --------------- | -------------- |----------- |
+| bge-m3                                                                              | √   | √        |√        |
+| bge-embedding                                                                       | √   | √        |√        |
+| bce-embedding                                                                       | √   | √        |√        |
+| puff                                                                                | √   | √        |√        |
+| piccolo-base-zh-embedding                                                           | √   | √        |√        |
+| acge_text_embedding                                                                 | √   | √        |√        |
+| Yinka                                                                               | √   | √        |√        |
+| zpoint_large_embedding_zh                                                           | √   | √        |√        |
+| xiaobu-embedding                                                                    | √   | √        |√        |
+| Conan-embedding-v1                                                                  | √   | √        |√        |
+| qwen3-embedding                                                                     | √   | √        |√        |
+| ritrieve_zh_v1                                                                      | √   | √        |√        |
+| jina-embeddings-v3                                                                  | √   | √        |√        |
+| KoalaAI/Text-Moderation（文本审核/多分类，审核文本是否存在暴力、色情等）                | ×   | √         |×        |
+| protectai/deberta-v3-base-prompt-injection-v2（提示注入/2分类，审核文本为提示注入）    | ×   | √         |×        |
+| bge-vl                                                                              | √   | ×        |×        |
+| jina-reranker-m0                                                                    | √   | ×        |×        |
 | bge-reranker                                                                        | √   | √        |×        |
-| bge-embedding                                                                       | √   | √        |×        |
 | bce-reranker                                                                        | √   | √        |×        |
-| bce-embedding                                                                       | √   | √        |×        |
-| puff                                                                                | √   | √        |×        |
-| piccolo-base-zh-embedding                                                           | √   | √        |×        |
-| acge_text_embedding                                                                 | √   | √        |×        |
-| Yinka                                                                               | √   | √        |×        |
-| zpoint_large_embedding_zh                                                           | √   | √        |×        |
-| xiaobu-embedding                                                                    | √   | √        |×        |
-| Conan-embedding-v1                                                                  | √   | √        |×        |
-| qwen3-embedding                                                                     | √   | √        |√        |
-| ritrieve_zh_v1                                                                      | √   | √        |×        |
-| KoalaAI/Text-Moderation（文本审核/多分类，审核文本是否存在暴力、色情等）                 | ×   | √        |×        |
-| protectai/deberta-v3-base-prompt-injection-v2（提示注入/2分类，审核文本为提示注入）    | ×   | √        |×        |
 
 目前 **ritrieve_zh_v1** C-MTEB榜单排行第一(MTEB: https://huggingface.co/spaces/mteb/leaderboard)
 

gpt_server/model_worker/embedding_vllm.py

@@ -7,8 +7,7 @@
 from infinity_emb.inference.select_model import get_engine_type_from_config
 from gpt_server.model_worker.base.model_worker_base import ModelWorkerBase
 from gpt_server.model_worker.utils import get_embedding_mode
-import torch
-import vllm
+import numpy as np
 from vllm import LLM
 
 label_to_category = {
@@ -71,6 +70,11 @@ async def get_embeddings(self, params):
             # ----------
             outputs = self.engine.embed(texts)
             embedding = [o.outputs.embedding for o in outputs]
+            embeddings_np = np.array(embedding)
+            # ------ L2归一化（沿axis=1，即对每一行进行归一化）-------
+            norm = np.linalg.norm(embeddings_np, ord=2, axis=1, keepdims=True)
+            normalized_embeddings_np = embeddings_np / norm
+            embedding = normalized_embeddings_np.tolist()
 
         ret["embedding"] = embedding
         return ret