vla.cpp是一种针对VLA模型的推理引擎。本文介绍部署的细节:

  1. 安装依赖
sudo apt install -y \
  libprotobuf-dev protobuf-compiler \
  libzmq3-dev
sudo apt install -y \
  libssl-dev \
  nlohmann-json3-dev
  1. 编译命令
rm -rf build
cmake -B build -DCMAKE_BUILD_TYPE=Release
cmake --build build -j$(nproc)
  1. 推断命令

以SmolVLA模型为例:

./build/vla-cli   --ckpt models/smolvla/smolvla-libero.gguf   --image assets/front.jpg --tokens 1,100,200,2 --pretty
  1. 权重转GGUF格式的命令
python scripts/convert_smolvla_to_gguf.py \
  --ckpt smolvla_base \
  --out models/smolvla_mushroom_base.gguf

smolvla_base 文件夹对应微调后得到的权重文件夹

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐