语言:
- 阿拉伯语
- 德语
- 英语
- 西班牙语
- 法语
- 印地语
- 意大利语
- 日语
- 荷兰语
- 葡萄牙语
- 俄语
- 瑞典语
- 土耳其语
- 乌克兰语
- 中文
许可证: mit
库名称: transformers
数据集:
- fixie-ai/librispeech_asr
- fixie-ai/common_voice_17_0
- fixie-ai/peoples_speech
- fixie-ai/gigaspeech
- fixie-ai/multilingual_librispeech
- fixie-ai/wenetspeech
- fixie-ai/covost2
评估指标:
- bleu
管道标签: 音频文本转文本
Ultravox 模型卡
Ultravox 是一个多模态语音大语言模型,基于预训练的 Llama3.1-70B-Instruct 和 whisper-large-v3-turbo 主干构建。
更多信息和 GitHub 仓库请访问 https://ultravox.ai。
模型详情
模型描述
Ultravox 是一个多模态模型,可以同时接收语音和文本作为输入(例如,文本系统提示和语音用户消息)。模型的输入是一个包含特殊 <|audio|>
伪标记的文本提示,模型处理器会将该标记替换为输入音频的嵌入表示。使用合并后的嵌入作为输入,模型将像往常一样生成输出文本。
在 Ultravox 的未来版本中,我们计划扩展标记词汇表以支持生成语义和声学音频标记,这些标记可以输入到声码器中以产生语音输出。当前版本的模型尚未应用偏好调优。
模型来源
- 仓库: https://ultravox.ai
- 演示: 见仓库
使用方法
可以将该模型视为一个既能听又能理解语音的大语言模型。因此,它可以作为语音代理使用,也可以用于语音到语音的翻译、语音音频分析等任务。
使用模型的方法如下:
import transformers
import numpy as np
import librosa
pipe = transformers.pipeline(model='fixie-ai/ultravox-v0_4_1-llama-3_1-70b', trust_remote_code=True)
path = "<输入音频路径>"
audio, sr = librosa.load(path, sr=16000)
turns = [
{
"role": "system",
"content": "你是一个友好且乐于助人的角色。你喜欢回答人们的问题。"
},
]
pipe({'audio': audio, 'turns': turns, 'sampling_rate': sr}, max_new_tokens=30)
训练详情
该模型使用了预训练的 Llama3.1-70B-Instruct 主干以及 whisper-large-v3-turbo 的编码器部分。
仅训练了多模态适配器,而 Whisper 编码器和 Llama 保持冻结状态。
我们使用了知识蒸馏损失,Ultravox 试图匹配基于文本的 Llama 主干的逻辑输出。
训练数据
训练数据集是多个自动语音识别(ASR)数据集的混合,并通过 Llama 3.1 8B 生成的延续进行了扩展,以及语音翻译数据集,这些数据集在翻译评估中带来了适度的改进。
训练过程
通过知识蒸馏进行监督式语音指令微调。更多信息,请参见 Ultravox 仓库中的训练代码。
训练超参数
- 训练方式: BF16 混合精度训练
- 使用的硬件: 8 块 H100 GPU
速度、大小和时间
当前版本的 Ultravox 在调用音频内容时,首次令牌生成时间(TTFT)约为 150 毫秒,使用 A100-40GB GPU 时,每秒生成令牌数约为 50-100,所有这些均基于 Llama 3.1 8B 主干。
访问 TheFastest.ai 的音频标签页,查看每日基准测试以及与其他现有模型的比较。
评估
|
Ultravox 0.4 70B |
Ultravox 0.4.1 70B |
英语-阿拉伯语 |
14.97 |
19.64 |
英语-德语 |
30.30 |
32.47 |
西班牙语-英语 |
39.55 |
40.76 |
俄语-英语 |
44.16 |
45.07 |
英语-加泰罗尼亚语 |
35.02 |
37.58 |
中文-英语 |
12.16 |
17.98 |