模型转换
- 测试可用的依赖版本如下:
torch==2.10.0
transformers==4.57.6
onnx==1.18.0
onnxruntime==1.22.0
einops==0.8.2
rknn-toolkit2==2.3.2
rkllm-toolkit==1.2.3
- 下载模型
从https://huggingface.co/openbmb/VoxCPM1.5下载模型,保存到./VoxCPM1.5文件夹。
- 转换模型
python scripts/build_rk3588_pipeline.py --model-dir VoxCPM1.5
转换后的模型会放置在build/rk3588/final_models/.
Model Conversion
- Tested dependency versions:
torch==2.10.0
transformers==4.57.6
onnx==1.18.0
onnxruntime==1.22.0
einops==0.8.2
rknn-toolkit2==2.3.2
rkllm-toolkit==1.2.3
- Download the model
Download the model from https://huggingface.co/openbmb/VoxCPM1.5 and save it to the ./VoxCPM1.5 directory.
- Convert the model
python scripts/build_rk3588_pipeline.py --model-dir VoxCPM1.5
The converted models will be placed in build/rk3588/final_models/.