xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 01:15:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] improve docs (#18734)

Browse Source 
Signed-off-by: reidliu41 <reid201711@gmail.com>
Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
Reid 2025-05-27 15:07:01 +08:00 committed by GitHub
parent 753944fa9b
commit fc6d0c290f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 57 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
examples/offline_inference/neuron_eagle.py
View File

@ -15,11 +15,13 @@ prompts = [
"What is annapurna labs?", "What is annapurna labs?",
] ]
# Create a sampling params object.
sampling_params = SamplingParams(top_k=1, max_tokens=500, ignore_eos=True)
# Create an LLM. def main():
llm = LLM( # Create a sampling params object.
sampling_params = SamplingParams(top_k=1, max_tokens=500, ignore_eos=True)
# Create an LLM.
llm = LLM(
model="/home/ubuntu/model_hf/Meta-Llama-3.1-70B-Instruct", model="/home/ubuntu/model_hf/Meta-Llama-3.1-70B-Instruct",
speculative_config={ speculative_config={
"model": "/home/ubuntu/model_hf/Llama-3.1-70B-Instruct-EAGLE-Draft", "model": "/home/ubuntu/model_hf/Llama-3.1-70B-Instruct-EAGLE-Draft",
@ -42,13 +44,17 @@ llm = LLM(
"enable_eagle_speculation": True, "enable_eagle_speculation": True,
"enable_fused_speculation": True, "enable_fused_speculation": True,
}, },
) )
# Generate texts from the prompts. The output is a list of RequestOutput objects # Generate texts from the prompts. The output is a list of RequestOutput objects
# that contain the prompt, generated text, and other information. # that contain the prompt, generated text, and other information.
outputs = llm.generate(prompts, sampling_params) outputs = llm.generate(prompts, sampling_params)
# Print the outputs. # Print the outputs.
for output in outputs: for output in outputs:
prompt = output.prompt prompt = output.prompt
generated_text = output.outputs[0].text generated_text = output.outputs[0].text
print(f"Prompt: {prompt!r}, \n\n\n\ Generated text: {generated_text!r}") print(f"Prompt: {prompt!r}, \n\n\n\ Generated text: {generated_text!r}")
if __name__ == "__main__":
main()

22
examples/offline_inference/qwen2_5_omni/README.md
View File

@ -6,14 +6,19 @@ This folder provides several example scripts on how to inference Qwen2.5-Omni of
```bash ```bash
# Audio + image + video # Audio + image + video
python examples/offline_inference/qwen2_5_omni/only_thinker.py -q mixed_modalities python examples/offline_inference/qwen2_5_omni/only_thinker.py \
-q mixed_modalities
# Read vision and audio inputs from a single video file # Read vision and audio inputs from a single video file
# NOTE: V1 engine does not support interleaved modalities yet. # NOTE: V1 engine does not support interleaved modalities yet.
VLLM_USE_V1=0 python examples/offline_inference/qwen2_5_omni/only_thinker.py -q use_audio_in_video VLLM_USE_V1=0 \
python examples/offline_inference/qwen2_5_omni/only_thinker.py \
-q use_audio_in_video
# Multiple audios # Multiple audios
VLLM_USE_V1=0 python examples/offline_inference/qwen2_5_omni/only_thinker.py -q multi_audios VLLM_USE_V1=0 \
python examples/offline_inference/qwen2_5_omni/only_thinker.py \
-q multi_audios
``` ```
This script will run the thinker part of Qwen2.5-Omni, and generate text response. This script will run the thinker part of Qwen2.5-Omni, and generate text response.
@ -22,11 +27,16 @@ You can also test Qwen2.5-Omni on a single modality:
```bash ```bash
# Process audio inputs # Process audio inputs
python examples/offline_inference/audio_language.py --model-type qwen2_5_omni python examples/offline_inference/audio_language.py \
--model-type qwen2_5_omni
# Process image inputs # Process image inputs
python examples/offline_inference/vision_language.py --modality image --model-type qwen2_5_omni python examples/offline_inference/vision_language.py \
--modality image \
--model-type qwen2_5_omni
# Process video inputs # Process video inputs
python examples/offline_inference/vision_language.py --modality video --model-type qwen2_5_omni python examples/offline_inference/vision_language.py \
--modality video \
--model-type qwen2_5_omni
``` ```