Audio-Text-to-Text
Transformers
Safetensors
step_audio_2
text-generation
audio-reasoning
chain-of-thought
multi-modal
step-audio-r1
custom_code
Instructions to use stepfun-ai/Step-Audio-R1.1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use stepfun-ai/Step-Audio-R1.1 with Transformers:
# Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("stepfun-ai/Step-Audio-R1.1", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
| license: apache-2.0 | |
| pipeline_tag: audio-text-to-text | |
| library_name: transformers | |
| tags: | |
| - audio-reasoning | |
| - chain-of-thought | |
| - multi-modal | |
| - step-audio-r1 | |
| ## Overview of Step-Audio-R1.1 | |
| <a href="https://www.stepfun.com/studio/audio?tab=conversation"><img src="https://img.shields.io/static/v1?label=Space%20Playground&message=Studio&color=yellow"></a> <a href="https://huggingface.co/spaces/stepfun-ai/Step-Audio-R1"><img src="https://img.shields.io/static/v1?label=Space&message=Web&color=green"></a>   | |
| ### Introduction | |
| Step-Audio R1.1 (Realtime) is a major upgrade to Step-Audio-R1, designed for interactive spoken dialogue with both **real-time responsiveness** and **strong reasoning capability**. | |
| Unlike conventional streaming speech models that trade intelligence for latency, R1.1 enables *thinking while speaking*, achieving high intelligence without sacrificing speed. | |
| ### Mind-Paced Speaking (Low Latency) | |
| Based on the research [*Mind-Paced Speaking*](MPS.pdf), the Realtime variant adopts a **Dual-Brain Architecture**: | |
| - A **Formulation Brain** responsible for high-level reasoning | |
| - An **Articulation Brain** dedicated to speech generation | |
| This decoupling allows the model to perform **Chain-of-Thought reasoning during speech output**, maintaining ultra-low latency while handling complex tasks in real time. | |
| ### Acoustic-Grounded Reasoning (High Intelligence) | |
| To address the *inverted scaling* issue鈥攚here reasoning over transcripts can degrade performance鈥擲tep-Audio R1.1 grounds its reasoning directly in acoustic representations rather than text alone. | |
| Through iterative self-distillation, extended deliberation becomes a strength instead of a liability. This enables effective test-time compute scaling and leads to **state-of-the-art performance**, including top-ranking results on the AA benchmark. | |
|  | |
|  | |
|  | |
| ## Online demonstration | |
| ### StepFun Audio Studio | |
| - Both Step-Audio-R1.1 are available in our [StepFun Audio Studio](https://www.stepfun.com/studio/audio). | |
| - You will need an API key from the [StepFun Open Platform](https://platform.stepfun.com/). | |
| ## WeChat group | |
| You can scan the following QR code to join our WeChat group for communication and discussion. | |
| <div align="center"> | |
| <img src="https://cdn-uploads.huggingface.co/production/uploads/66518fd07d8cb2629a514c18/DRdnp1SN-yxhlNOfy26mE.jpeg" width="200" alt="QR code"> | |
| </div> | |
| ## Model Usage | |
| ### 馃摐 Requirements | |
| - **GPU**: NVIDIA GPUs with CUDA support (tested on 4脳L40S/H100/H800/H20). | |
| - **Operating System**: Linux. | |
| - **Python**: >= 3.10.0. | |
| ### 猬囷笍 Download Model | |
| First, you need to download the Step-Audio-R1 model weights. | |
| **Method A 路 Git LFS** | |
| ```bash | |
| git lfs install | |
| git clone https://huggingface.co/stepfun-ai/Step-Audio-R1.1 | |
| ``` | |
| **Method B 路 Hugging Face CLI** | |
| ```bash | |
| hf download stepfun-ai/Step-Audio-R1.1 --local-dir ./Step-Audio-R1.1 | |
| ``` | |
| ### 馃殌 Deployment and Execution | |
| We provide two ways to serve the model: Docker (recommended) or compiling the customized vLLM backend. | |
| #### 馃惓 Method 1 路 Run with Docker (Recommended) | |
| A customized vLLM image is required. | |
| 1. **Pull the image**: | |
| ```bash | |
| docker pull stepfun2025/vllm:step-audio-2-v20250909 | |
| ``` | |
| 2. **Start the service**: | |
| Assuming the model is downloaded in the `Step-Audio-R1` folder in the current directory. | |
| ```bash | |
| docker run --rm -ti --gpus all \ | |
| -v $(pwd)/Step-Audio-R1.1:/Step-Audio-R1.1 \ | |
| -p 9999:9999 \ | |
| stepfun2025/vllm:step-audio-2-v20250909 \ | |
| -- vllm serve /Step-Audio-R1.1 \ | |
| --served-model-name Step-Audio-R1.1 \ | |
| --port 9999 \ | |
| --max-model-len 16384 \ | |
| --max-num-seqs 32 \ | |
| --tensor-parallel-size 4 \ | |
| --chat-template '{%- macro render_content(content) -%}{%- if content is string -%}{{- content.replace("<audio_patch>\n", "<audio_patch>") -}}{%- elif content is mapping -%}{{- content['"'"'value'"'"'] if '"'"'value'"'"' in content else content['"'"'text'"'"'] -}}{%- elif content is iterable -%}{%- for item in content -%}{%- if item.type == '"'"'text'"'"' -%}{{- item['"'"'value'"'"'] if '"'"'value'"'"' in item else item['"'"'text'"'"'] -}}{%- elif item.type == '"'"'audio'"'"' -%}<audio_patch>{%- endif -%}{%- endfor -%}{%- endif -%}{%- endmacro -%}{%- if tools -%}{{- '"'"'<|BOT|>system\n'"'"' -}}{%- if messages[0]['"'"'role'"'"'] == '"'"'system'"'"' -%}{{- render_content(messages[0]['"'"'content'"'"']) + '"'"'<|EOT|>'"'"' -}}{%- endif -%}{{- '"'"'<|BOT|>tool_json_schemas\n'"'"' + tools|tojson + '"'"'<|EOT|>'"'"' -}}{%- else -%}{%- if messages[0]['"'"'role'"'"'] == '"'"'system'"'"' -%}{{- '"'"'<|BOT|>system\n'"'"' + render_content(messages[0]['"'"'content'"'"']) + '"'"'<|EOT|>'"'"' -}}{%- endif -%}{%- endif -%}{%- for message in messages -%}{%- if message["role"] == "user" -%}{{- '"'"'<|BOT|>human\n'"'"' + render_content(message["content"]) + '"'"'<|EOT|>'"'"' -}}{%- elif message["role"] == "assistant" -%}{{- '"'"'<|BOT|>assistant\n'"'"' + (render_content(message["content"]) if message["content"] else '"'"''"'"') -}}{%- set is_last_assistant = true -%}{%- for m in messages[loop.index:] -%}{%- if m["role"] == "assistant" -%}{%- set is_last_assistant = false -%}{%- endif -%}{%- endfor -%}{%- if not is_last_assistant -%}{{- '"'"'<|EOT|>'"'"' -}}{%- endif -%}{%- elif message["role"] == "function_output" -%}{%- else -%}{%- if not (loop.first and message["role"] == "system") -%}{{- '"'"'<|BOT|>'"'"' + message["role"] + '"'"'\n'"'"' + render_content(message["content"]) + '"'"'<|EOT|>'"'"' -}}{%- endif -%}{%- endif -%}{%- endfor -%}{%- if add_generation_prompt -%}{{- '"'"'<|BOT|>assistant\n<think>\n'"'"' -}}{%- endif -%}' \ | |
| --enable-log-requests \ | |
| --interleave-mm-strings \ | |
| --trust-remote-code | |
| ``` | |
| After the service starts, it will listen on `localhost:9999`. | |
| #### 馃惓 Method 2 路 Run from Source (Compile vLLM) | |
| Step-Audio-R1 requires a customized vLLM backend. | |
| 1. **Download Source Code**: | |
| ```bash | |
| git clone https://github.com/stepfun-ai/vllm.git | |
| cd vllm | |
| ``` | |
| 2. **Prepare Environment**: | |
| ```bash | |
| python3 -m venv .venv | |
| source .venv/bin/activate | |
| ``` | |
| 3. **Install and Compile**: | |
| vLLM contains both C++ and Python code. We mainly modified the Python code, so the C++ part can use the pre-compiled version to speed up the process. | |
| ```bash | |
| # Use pre-compiled C++ extensions (Recommended) | |
| VLLM_USE_PRECOMPILED=1 pip install -e . | |
| ``` | |
| 4. **Switch Branch**: | |
| After compilation, switch to the branch that supports Step-Audio. | |
| ```bash | |
| git checkout feat/step-audio-support | |
| ``` | |
| 5. **Start the Service**: | |
| ```bash | |
| # Ensure you are in the vllm directory and the virtual environment is activated | |
| source .venv/bin/activate | |
| python3 -m vllm.entrypoints.openai.api_server \ | |
| --model ../Step-Audio-R1.1 \ | |
| --served-model-name Step-Audio-R1.1 \ | |
| --port 9999 \ | |
| --host 0.0.0.0 \ | |
| --max-model-len 65536 \ | |
| --max-num-seqs 128 \ | |
| --tensor-parallel-size 4 \ | |
| --gpu-memory-utilization 0.85 \ | |
| --trust-remote-code \ | |
| --enable-log-requests \ | |
| --interleave-mm-strings \ | |
| --chat-template '{%- macro render_content(content) -%}{%- if content is string -%}{{- content.replace("<audio_patch>\n", "<audio_patch>") -}}{%- elif content is mapping -%}{{- content['"'"'value'"'"'] if '"'"'value'"'"' in content else content['"'"'text'"'"'] -}}{%- elif content is iterable -%}{%- for item in content -%}{%- if item.type == '"'"'text'"'"' -%}{{- item['"'"'value'"'"'] if '"'"'value'"'"' in item else item['"'"'text'"'"'] -}}{%- elif item.type == '"'"'audio'"'"' -%}<audio_patch>{%- endif -%}{%- endfor -%}{%- endif -%}{%- endmacro -%}{%- if tools -%}{{- '"'"'<|BOT|>system\n'"'"' -}}{%- if messages[0]['"'"'role'"'"'] == '"'"'system'"'"' -%}{{- render_content(messages[0]['"'"'content'"'"']) + '"'"'<|EOT|>'"'"' -}}{%- endif -%}{{- '"'"'<|BOT|>tool_json_schemas\n'"'"' + tools|tojson + '"'"'<|EOT|>'"'"' -}}{%- else -%}{%- if messages[0]['"'"'role'"'"'] == '"'"'system'"'"' -%}{{- '"'"'<|BOT|>system\n'"'"' + render_content(messages[0]['"'"'content'"'"']) + '"'"'<|EOT|>'"'"' -}}{%- endif -%}{%- endif -%}{%- for message in messages -%}{%- if message["role"] == "user" -%}{{- '"'"'<|BOT|>human\n'"'"' + render_content(message["content"]) + '"'"'<|EOT|>'"'"' -}}{%- elif message["role"] == "assistant" -%}{{- '"'"'<|BOT|>assistant\n'"'"' + (render_content(message["content"]) if message["content"] else '"'"''"'"') -}}{%- set is_last_assistant = true -%}{%- for m in messages[loop.index:] -%}{%- if m["role"] == "assistant" -%}{%- set is_last_assistant = false -%}{%- endif -%}{%- endfor -%}{%- if not is_last_assistant -%}{{- '"'"'<|EOT|>'"'"' -}}{%- endif -%}{%- elif message["role"] == "function_output" -%}{%- else -%}{%- if not (loop.first and message["role"] == "system") -%}{{- '"'"'<|BOT|>'"'"' + message["role"] + '"'"'\n'"'"' + render_content(message["content"]) + '"'"'<|EOT|>'"'"' -}}{%- endif -%}{%- endif -%}{%- endfor -%}{%- if add_generation_prompt -%}{{- '"'"'<|BOT|>assistant\n<think>\n'"'"' -}}{%- endif -%}' | |
| ``` | |
| After the service starts, it will listen on `localhost:9999`. |