SQL-GPT / README.md
nobodynosql's picture
Update README.md
ccc0474 verified
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

{
        "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": "What are the id, name, price and color of the products which have not been ordered for at least twice? | products : products.product_price , products.product_color , products.product_id , products.product_name , products.parent_product_id | order_items : order_items.product_id , order_items.order_id , order_items.order_item_id , order_items.order_item_status_code | orders : orders.order_id , orders.customer_id , orders.order_status_code , orders.date_order_placed | customers : customers.county , customers.country , customers.customer_id , customers.gender_code , customers.customer_first_name | orders.customer_id = customers.customer_id | order_items.order_id = orders.order_id | order_items.product_id = products.product_id",
                "chat_round_id": 1
            },
            {
                "role": "bot",
                "content": " select product_id , product_name , product_price , product_color from products except select products.product_id , products.product_name , products.product_price , products.product_color from products join order_items on products.product_id = order_items.product_id join orders on order_items.order_id = orders.order_id group by products.product_id having count ( * ) >= 2",
                "chat_round_id": 2
            }
        ]
}

Framework versions

  • PEFT 0.7.0

Base Model