Skip to main content

No project description provided

Project description

Xiaomi-MiMo

━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
MiMo Audio: Audio Language Models are Few-Shot Learners
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━



Introduction

Existing audio language models typically rely on task-specific fine-tuning to accomplish particular audio tasks. In contrast, humans are able to generalize to new audio tasks with only a few examples or simple instructions. GPT-3 has shown that scaling next-token prediction pretraining enables strong generalization capabilities in text, and we believe this paradigm is equally applicable to the audio domain. By scaling MiMo-Audio's pretraining data to over one hundred million of hours, we observe the emergence of few-shot learning capabilities across a diverse set of audio tasks. We develop a systematic evaluation of these capabilities and find that MiMo-Audio-7B-Base achieves SOTA performance on both speech intelligence and audio understanding benchmarks among open-source models. Beyond standard metrics, MiMo-Audio-7B-Base generalizes to tasks absent from its training data, such as voice conversion, style transfer, and speech editing. MiMo-Audio-7B-Base also demonstrates powerful speech continuation capabilities, capable of generating highly realistic talk shows, recitations, livestreaming and debates. At the post-training stage, we curate a diverse instruction-tuning corpus and introduce thinking mechanisms into both audio understanding and generation. MiMo-Audio-7B-Instruct achieves open-source SOTA on audio understanding benchmarks, spoken dialogue benchmarks and instruct-TTS evaluations, approaching or surpassing closed-source models.

Results

Architecture

MiMo-Audio-Tokenizer

MiMo-Audio-Tokenizer is a 1.2B-parameter Transformer operating at 25 Hz. It employs an eight-layer RVQ stack to generate 200 tokens per second. By jointly optimizing semantic and reconstruction objectives, we train MiMo-Audio-Tokenizer from scratch on a 10-million-hour corpus, achieving superior reconstruction quality and facilitating downstream language modeling.

Tokenizer

MiMo-Audio couples a patch encoder, an LLM, and a patch decoder to improve modeling efficiency for high-rate sequences and bridge the length mismatch between speech and text. The patch encoder aggregates four consecutive time steps of RVQ tokens into a single patch, downsampling the sequence to a 6.25 Hz representation for the LLM. The patch decoder autoregressively generates the full 25 Hz RVQ token sequence via a delayed-generation scheme.

MiMo-Audio

Arch

Explore MiMo-Audio Now! 🚀🚀🚀

Model Download

Models 🤗 Hugging Face
MiMo-Audio-Tokenizer XiaomiMiMo/MiMo-Audio-Tokenizer
MiMo-Audio-7B-Base XiaomiMiMo/MiMo-Audio-7B-Base
MiMo-Audio-7B-Instruct XiaomiMiMo/MiMo-Audio-7B-Instruct

Getting Started

Spin up the MiMo-Audio demo in minutes with the built-in Gradio app.

Installation

git clone https://github.com/XiaomiMiMo/MiMo-Audio.git
cd MiMo-Audio
pip install -e .

Run the demo

python run_mimo_audio.py

This launches a local Gradio interface where you can try MiMo-Audio interactively.

Demo UI

Enter the local paths for MiMo-Audio-Tokenizer and MiMo-Audio-7B-Instruct, then enjoy the full functionality of MiMo-Audio!

Inference Scripts

Base Model

We provide an example script to explore the in-context learning capabilities of MiMo-Audio-7B-Base.
See: inference_example_pretrain.py

Instruct Model

To try the instruction-tuned model MiMo-Audio-7B-Instruct, use the corresponding inference script.
See: inference_example_sft.py

Evaluation Toolkit

Full evaluation suite are available at 🌐MiMo-Audio-Eval.

This toolkit is designed to evaluate MiMo-Audio and other recent audio LLMs as mentioned in the paper. It provides a flexible and extensible framework, supporting a wide range of datasets, tasks, and models.

Citation

@misc{coreteam2025mimoaudio,
      title={MiMo-Audio: Audio Language Models are Few-Shot Learners}, 
      author={LLM-Core-Team Xiaomi},
      year={2025},
      url={https://github.com/XiaomiMiMo/MiMo-Audio}, 
}

Contact

Please contact us at mimo@xiaomi.com or open an issue if you have any questions.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tts_webui_mimo_audio-0.0.0.tar.gz (44.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tts_webui_mimo_audio-0.0.0-py3-none-any.whl (44.3 kB view details)

Uploaded Python 3

File details

Details for the file tts_webui_mimo_audio-0.0.0.tar.gz.

File metadata

  • Download URL: tts_webui_mimo_audio-0.0.0.tar.gz
  • Upload date:
  • Size: 44.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.11

File hashes

Hashes for tts_webui_mimo_audio-0.0.0.tar.gz
Algorithm Hash digest
SHA256 9026dd17791e3d39688f2a257726ff1b254e5f274514dc0e9536d3f77e9a0ed4
MD5 4f6c10a823096c3df9006afa8373ab37
BLAKE2b-256 93b3a37ce98a1b839c25a1bf2edffdaaca254d914430e37a5307c7a1b9b948e8

See more details on using hashes here.

File details

Details for the file tts_webui_mimo_audio-0.0.0-py3-none-any.whl.

File metadata

File hashes

Hashes for tts_webui_mimo_audio-0.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 96bedd1e1cf9d05883628a7318d14caeac946df1d7eb80c88ef00a9199b74af5
MD5 86089105b17f6ed5d62c2cd57b021557
BLAKE2b-256 981ca5921ff6a9f7276986d0933fdccb18f333435dcdfd8659725a76f1d6fc81

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page