2.6 KiB
Qwen2.5-Omni: Offline inference
Source https://github.com/vllm-project/vllm-omni/tree/main/examples/offline_inference/qwen2_5_omni.
Setup
See the pipeline and deploy configuration documentation to configure memory allocation for your hardware.
Run examples
Multiple Prompts
Get into the example folder
cd examples/offline_inference/qwen2_5_omni
Then run the command below. Note: for processing large volume data, it uses py_generator mode, which will return a python generator from Omni class.
bash run_multiple_prompts.sh
Single Prompt
Get into the example folder
cd examples/offline_inference/qwen2_5_omni
Then run the command below.
bash run_single_prompt.sh
Modality control
If you want to control output modalities, e.g. only output text, you can run the command below:
python end2end.py --output-wav output_audio \
--query-type mixed_modalities \
--modalities text
Using Local Media Files
The end2end.py script supports local media files (audio, video, image) via CLI arguments:
# Use single local media files
python end2end.py --query-type use_image --image-path /path/to/image.jpg
python end2end.py --query-type use_video --video-path /path/to/video.mp4
python end2end.py --query-type use_audio --audio-path /path/to/audio.wav
# Combine multiple local media files
python end2end.py --query-type mixed_modalities \
--video-path /path/to/video.mp4 \
--image-path /path/to/image.jpg \
--audio-path /path/to/audio.wav
# Use audio from video file
python end2end.py --query-type use_audio_in_video --video-path /path/to/video.mp4
If media file paths are not provided, the script will use default assets. Supported query types:
use_image: Image input onlyuse_video: Video input onlyuse_audio: Audio input onlymixed_modalities: Audio + image + videouse_audio_in_video: Extract audio from videotext: Text-only query
Example materials
??? abstract "end2end.py"
py --8<-- "examples/offline_inference/qwen2_5_omni/end2end.py"
??? abstract "extract_prompts.py"
py --8<-- "examples/offline_inference/qwen2_5_omni/extract_prompts.py"
??? abstract "run_multiple_prompts.sh"
sh --8<-- "examples/offline_inference/qwen2_5_omni/run_multiple_prompts.sh"
??? abstract "run_single_prompt.sh"
sh --8<-- "examples/offline_inference/qwen2_5_omni/run_single_prompt.sh"