OmniSQL是一个基于代码的工具,需要用户具备一定的编程基础。以下是完整的安装使用步骤:
- 环境准备:确保安装Python 3.8或更高版本,可通过命令行输入
python --version
检查 - 下载项目:可以从GitHub直接下载ZIP压缩包或使用Git命令克隆项目
- 安装依赖:进入项目目录运行
pip install -r requirements.txt
- 额外安装:如需模型推理,需安装vLLM或Transformers
- 模型下载:需从HuggingFace下载所需规模的模型和SynSQL-2.5M数据集
- 运行检查:进入目录运行
python omnisql.py
检查环境
使用准备时需注意:
- 硬件要求:7B模型约需14GB显存,32B需要更高配置
- 当前仅支持SQLite数据库
- 项目examples文件夹提供多种提示模板示例
本答案来源于文章《OmniSQL:将自然语言转化为高质量SQL查询的模型》