SQL-GPT / README.md
nobodynosql's picture
Update README.md
d38d82b verified
|
raw
history blame
1.03 kB
metadata
license: apache-2.0

Model Details

Model Description

This model is a fine-tuned version based on deepseek, and after fine-tuning, it can achieve a score of 84.0 on the Spider dev leaderboard, significantly surpassing the original capabilities of the deepseek model. This model is sql-gen model, it need use result of schema linking model to get most related columns.

Data Format

{
    "data_name":"spider",
    "id":0,
    "db_id":"farm",
    "chat_rounds":[
        {
            "role":"system",
            "content":"You are a professional SQL engineer and you are writing SQL queries for data query tasks.\n",
            "chat_round_id":0
        },
        {
            "role":"human",
            "content":"Question | table1 : column1 , column2 | ... | tablen: columnn, columnn+1 | foreign_keys",
            "chat_round_id":1
        },
        {
            "role":"bot",
            "content":" sql",
            "chat_round_id":2
        }
    ]
}

Framework versions

  • PEFT 0.7.0