Views
No views yet
1uv add mlx-lm-omni
2# or
3uv add https://github.com/giangndm/mlx-lm-omni.git1from mlx_lm_omni import load, generate
2import librosa
3from io import BytesIO
4from urllib.request import urlopen
5
6model, tokenizer = load("giangndm/qwen2.5-omni-7b-mlx-4bit")
7
8audio_path = "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-Audio/audio/1272-128104-0000.flac"
9audio = librosa.load(BytesIO(urlopen(audio_path).read()), sr=16000)[0]
10
11messages = [
12 {"role": "system", "content": "You are a speech recognition model."},
13 {"role": "user", "content": "Transcribe the English audio into text without any punctuation marks.", "audio": audio},
14]
15prompt = tokenizer.apply_chat_template(
16 messages, add_generation_prompt=True
17)
18
19text = generate(model, tokenizer, prompt=prompt, verbose=True)