dnrops
/
sqlcoder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Rishabh Srivastava 2024-02-11 10:49:51 +05:30 committed by GitHub
parent 74fba70d17
commit 633b9b5ff5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -6,13 +6,14 @@ Defog's SQLCoder is a family of state-of-the-art LLMs for converting natural lan
## TL;DR
SQLCoder is a family of large language models that outperforms `gpt-4` and `gpt-4-turbo` for natural language to SQL generation tasks on our [sql-eval](https://github.com/defog-ai/sql-eval) framework, and significantly outperform all popular open-source models.
![Percentage of correctly generated SQL queries on novel schemas not seen in training (n = 200) in SQL-Eval](https://github.com/defog-ai/sqlcoder/assets/5008293/839aa98f-101d-4baa-8e80-bb4fd5110012)]
![Percentage of correctly generated SQL queries on novel schemas not seen in training (n = 200), with 4 beams (2)](https://github.com/defog-ai/sqlcoder/assets/5008293/22b891db-2201-4b30-a52d-22376ba8ec86)
## Installing SQLCoder
If running on a device with an NVIDIA GPU with more than 16GB VRAM (best performance)
`pip install "sqlcoder[transformers]"`
If running on Apple Silicon (medium performance)
If running on Apple Silicon (less good performance, because of quantization and lack of beam search)
`CMAKE_ARGS="-DLLAMA_METAL=on" pip install "sqlcoder[llama-cpp]"`
SQLCoder has not been tested on other platforms yet. Contributions for testing on other platforms are very welcome!