File size: 994 Bytes
efc006c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | ---
license: other
base_model: Qwen/Qwen2.5-Coder-7B-Instruct
tags: [text-to-sql, bird, spider, qlora, aegis-sql]
---
# Qwen25-Coder7B-Aegis-Text2SQL
QLoRA fine-tune of `Qwen/Qwen2.5-Coder-7B-Instruct` for Text-to-SQL, trained as the local-path
generator of the AEGIS-SQL hybrid system (three-axis constrained NL2SQL:
accuracy / cost / privacy).
- **Data**: 1,000 curated examples — 500 BIRD-train + 500 diversity-stratified
Spider-train (set ops, HAVING, nested, multi-join rebalanced); splits 798/101/101.
- **Recipe**: 4-bit NF4 QLoRA, r=32 alpha=64, lr=0.0001,
3 epochs, completion-only loss, max_seq_len=4096, seed=42.
- **Prompt format**: chat messages; user = `Database:` + schema (Tables + Foreign
Keys) + `Question:` + `Return only SQL.`; assistant = bare SQL.
- **Quick proxy on held-out test**: see `quick_eval.json` in this repo.
Full BIRD EX requires the official evaluator + databases.
Trained with the text2sql_aegis_experiment package (notebooks/qwen_aegis.ipynb).
|