diff --git a/docs/source/tutorials/Qwen3-VL-Embedding.md b/docs/source/tutorials/Qwen3-VL-Embedding.md new file mode 100644 index 00000000..d39aed9c --- /dev/null +++ b/docs/source/tutorials/Qwen3-VL-Embedding.md @@ -0,0 +1,127 @@ +# Qwen3-VL-Embedding + +## Introduction + +The Qwen3-VL-Embedding and Qwen3-VL-Reranker model series are the latest additions to the Qwen family, built upon the recently open-sourced and powerful Qwen3-VL foundation model. Specifically designed for multimodal information retrieval and cross-modal understanding, this suite accepts diverse inputs including text, images, screenshots, and videos, as well as inputs containing a mixture of these modalities. This guide describes how to run the model with vLLM Ascend. + +## Supported Features + +Refer to [supported features](../user_guide/support_matrix/supported_models.md) to get the model's supported feature matrix. + +## Environment Preparation + +### Model Weight + +- `Qwen3-VL-Embedding-8B` [Download model weight](https://www.modelscope.cn/models/Qwen/Qwen3-VL-Embedding-8B) +- `Qwen3-VL-Embedding-2B` [Download model weight](https://www.modelscope.cn/models/Qwen/Qwen3-VL-Embedding-2B) + +It is recommended to download the model weight to the shared directory of multiple nodes, such as `/root/.cache/` + +### Installation + +You can use our official docker image to run `Qwen3-VL-Embedding` series models. + +- Start the docker image on your node, refer to [using docker](../installation.md#set-up-using-docker). + +If you don't want to use the docker image as above, you can also build all from source: + +- Install `vllm-ascend` from source, refer to [installation](../installation.md). + +## Deployment + +Using the Qwen3-VL-Embedding-8B model as an example, first run the docker container with the following command: + +### Online Inference + +```bash +vllm serve Qwen/Qwen3-VL-Embedding-8B --runner pooling +``` + +Once your server is started, you can query the model with input prompts. + +```bash +curl http://127.0.0.1:8000/v1/embeddings -H "Content-Type: application/json" -d '{ + "input": [ + "The capital of China is Beijing.", + "Gravity is a force that attracts two bodies towards each other. It gives weight to physical objects and is responsible for the movement of planets around the sun." + ] +}' +``` + +### Offline Inference + +```python +import torch +from vllm import LLM + +def get_detailed_instruct(task_description: str, query: str) -> str: + return f'Instruct: {task_description}\nQuery: {query}' + + +if __name__=="__main__": + # Each query must come with a one-sentence instruction that describes the task + task = 'Given a web search query, retrieve relevant passages that answer the query' + + queries = [ + get_detailed_instruct(task, 'What is the capital of China?'), + get_detailed_instruct(task, 'Explain gravity') + ] + # No need to add instruction for retrieval documents + documents = [ + "The capital of China is Beijing.", + "Gravity is a force that attracts two bodies towards each other. It gives weight to physical objects and is responsible for the movement of planets around the sun." + ] + input_texts = queries + documents + + model = LLM(model="Qwen/Qwen3-VL-Embedding-8B", + runner="pooling", + distributed_executor_backend="mp") + + outputs = model.embed(input_texts) + embeddings = torch.tensor([o.outputs.embedding for o in outputs]) + scores = (embeddings[:2] @ embeddings[2:].T) + print(scores.tolist()) +``` + +If you run this script successfully, you can see the info shown below: + +```bash +Adding requests: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 4/4 [00:00<00:00, 192.47it/s] +Processed prompts: 0%| | 0/4 [00:00system +Judge whether the Document meets the requirements based on the Query and the Instruct provided. Note that the answer can only be "yes" or "no".<|im_end|> +<|im_start|>user +: {{ + messages + | selectattr("role", "eq", "system") + | map(attribute="content") + | first + | default("Given a search query, retrieve relevant candidates that answer the query.") +}}:{{ + messages + | selectattr("role", "eq", "query") + | map(attribute="content") + | first +}} +:{{ + messages + | selectattr("role", "eq", "document") + | map(attribute="content") + | first +}}<|im_end|> +<|im_start|>assistant + +``` + +Save this file to a location of your choice (e.g., `./qwen3_vl_reranker.jinja`). + +### Online Inference + +Start the server with the following command: + +```bash +vllm serve Qwen/Qwen3-VL-Reranker-8B \ + --runner pooling \ + --max-model-len 4096 \ + --hf_overrides '{"architectures": ["Qwen3VLForSequenceClassification"],"classifier_from_token": ["no", "yes"],"is_original_qwen3_reranker": true}' \ + --chat-template ./qwen3_vl_reranker.jinja +``` + +Once your server is started, you can send request with follow examples. + +```python +import requests + +url = "http://127.0.0.1:8000/v1/rerank" + +# Please use the query_template and document_template to format the query and +# document for better reranker results. + +prefix = '<|im_start|>system\nJudge whether the Document meets the requirements based on the Query and the Instruct provided. Note that the answer can only be "yes" or "no".<|im_end|>\n<|im_start|>user\n' +suffix = "<|im_end|>\n<|im_start|>assistant\n" + +query_template = "{prefix}: {instruction}\n: {query}\n" +document_template = ": {doc}{suffix}" + +instruction = ( + "Given a search query, retrieve relevant candidates that answer the query." +) + +query = "What is the capital of China?" + +documents = [ + "The capital of China is Beijing.", + "Gravity is a force that attracts two bodies towards each other. It gives weight to physical objects and is responsible for the movement of planets around the sun.", +] + +documents = [ + document_template.format(doc=doc, suffix=suffix) for doc in documents +] + +response = requests.post(url, + json={ + "query": query_template.format(prefix=prefix, instruction=instruction, query=query), + "documents": documents, + }).json() + +print(response) +``` + +If you run this script successfully, you will see a list of scores printed to the console, similar to this: + +```bash +{'id': 'rerank-ac3495afa8e12404', 'model': 'Qwen/Qwen3-VL-Reranker-8B', 'usage': {'prompt_tokens': 315, 'total_tokens': 315}, 'results': [{'index': 0, 'document': {'text': ': The capital of China is Beijing.<|im_end|>\n<|im_start|>assistant\n', 'multi_modal': None}, 'relevance_score': 0.6368980407714844}, {'index': 1, 'document': {'text': ': Gravity is a force that attracts two bodies towards each other. It gives weight to physical objects and is responsible for the movement of planets around the sun.<|im_end|>\n<|im_start|>assistant\n', 'multi_modal': None}, 'relevance_score': 0.20816077291965485}]} +``` + +### Offline Inference + +```python +from vllm import LLM + +model_name = "Qwen/Qwen3-VL-Reranker-8B" + +# What is the difference between the official original version and one +# that has been converted into a sequence classification model? +# Qwen3-Reranker is a language model that doing reranker by using the +# logits of "no" and "yes" tokens. +# It needs to computing 151669 tokens logits, making this method extremely +# inefficient, not to mention incompatible with the vllm score API. +# A method for converting the original model into a sequence classification +# model was proposed. See: https://huggingface.co/Qwen/Qwen3-Reranker-0.6B/discussions/3 +# Models converted offline using this method can not only be more efficient +# and support the vllm score API, but also make the init parameters more +# concise, for example. +# model = LLM(model="Qwen/Qwen3-VL-Reranker-8B", runner="pooling") + +# If you want to load the official original version, the init parameters are +# as follows. + +model = LLM( + model=model_name, + runner="pooling", + hf_overrides={ + # Manually route to sequence classification architecture + # This tells vLLM to use Qwen3VLForSequenceClassification instead of + # the default Qwen3VLForConditionalGeneration + "architectures": ["Qwen3VLForSequenceClassification"], + # Specify which token logits to extract from the language model head + # The original reranker uses "no" and "yes" token logits for scoring + "classifier_from_token": ["no", "yes"], + # Enable special handling for original Qwen3-Reranker models + # This flag triggers conversion logic that transforms the two token + # vectors into a single classification vector + "is_original_qwen3_reranker": True, + }, +) + +# Why do we need hf_overrides for the official original version: +# vllm converts it to Qwen3VLForSequenceClassification when loaded for +# better performance. +# - Firstly, we need using `"architectures": ["Qwen3VLForSequenceClassification"],` +# to manually route to Qwen3VLForSequenceClassification. +# - Then, we will extract the vector corresponding to classifier_from_token +# from lm_head using `"classifier_from_token": ["no", "yes"]`. +# - Third, we will convert these two vectors into one vector. The use of +# conversion logic is controlled by `using "is_original_qwen3_reranker": True`. + +# Please use the query_template and document_template to format the query and +# document for better reranker results. + +prefix = '<|im_start|>system\nJudge whether the Document meets the requirements based on the Query and the Instruct provided. Note that the answer can only be "yes" or "no".<|im_end|>\n<|im_start|>user\n' +suffix = "<|im_end|>\n<|im_start|>assistant\n" + +query_template = "{prefix}: {instruction}\n: {query}\n" +document_template = ": {doc}{suffix}" + +if __name__ == "__main__": + instruction = ( + "Given a search query, retrieve relevant candidates that answer the query." + ) + + query = "What is the capital of China?" + + documents = [ + "The capital of China is Beijing.", + "Gravity is a force that attracts two bodies towards each other. It gives weight to physical objects and is responsible for the movement of planets around the sun.", + ] + + documents = [document_template.format(doc=doc, suffix=suffix) for doc in documents] + + outputs = model.score(query_template.format(prefix=prefix, instruction=instruction, query=query), documents) + + print([output.outputs.score for output in outputs]) +``` + +If you run this script successfully, you will see a list of scores printed to the console, similar to this: + +```bash +Adding requests: 100%|████████████████████████████████████████████████████████████████████████████████████████| 2/2 [00:00<00:00, 2409.83it/s] +Processed prompts: 0%| | 0/2 [00:00