Ovis的安装需要以下关键步骤:
- 环境准备:需预先安装Git和Anaconda,创建Python 3.10的conda虚拟环境
- 代码获取:通过
git clone git@github.com:AIDC-AI/Ovis.git
克隆仓库 - 依赖安装:使用requirements.txt安装基础依赖包,建议额外安装Flash Attention加速库
- 验证环节:需确认PyTorch版本(推荐2.4.0)和CUDA环境
运行方式提供两种选择:
- 命令行模式:需编写Python脚本加载模型并处理输入输出
- Gradio界面:通过
python ovis/serve/server.py
启动可视化服务
注意:不同参数规模的模型对显存有不同要求,1B模型至少需要4GB显存。
本答案来源于文章《Ovis:视觉与文本对齐模型,精准反推图像提示词》