Card: link base model Darwin-180B-RSI and weight download note
Browse files
README.md
CHANGED
|
@@ -1,75 +1,86 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
language:
|
| 4 |
-
- en
|
| 5 |
-
tags:
|
| 6 |
-
- text-to-sql
|
| 7 |
-
- nl2sql
|
| 8 |
-
- bird
|
| 9 |
-
- sqlite
|
| 10 |
-
- darwin
|
| 11 |
-
pipeline_tag: text-generation
|
| 12 |
-
---
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
<
|
| 19 |
-
<a href="https://huggingface.co/FINAL-Bench"><img src="https://img.shields.io/badge/
|
| 20 |
-
</
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
|
| 44 |
-
|
|
| 45 |
-
|
|
| 46 |
-
|
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
|
| 56 |
-
|
|
| 57 |
-
|
|
| 58 |
-
|
|
| 59 |
-
|
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
tags:
|
| 6 |
+
- text-to-sql
|
| 7 |
+
- nl2sql
|
| 8 |
+
- bird
|
| 9 |
+
- sqlite
|
| 10 |
+
- darwin
|
| 11 |
+
pipeline_tag: text-generation
|
| 12 |
+
base_model: FINAL-Bench/Darwin-180B-RSI
|
| 13 |
+
---
|
| 14 |
+
|
| 15 |
+
# Darwin-180B-NL2SQL
|
| 16 |
+
|
| 17 |
+
<p align="center">
|
| 18 |
+
<img src="https://img.shields.io/badge/BIRD_Mini--Dev_SQLite-83.80_EX-gold?style=for-the-badge" alt="BIRD Mini-Dev SQLite 83.80">
|
| 19 |
+
<a href="https://huggingface.co/FINAL-Bench/Darwin-180B-RSI"><img src="https://img.shields.io/badge/Model-Darwin--180B--RSI-blue?style=for-the-badge" alt="Darwin-180B-RSI"></a>
|
| 20 |
+
<a href="https://huggingface.co/FINAL-Bench"><img src="https://img.shields.io/badge/Org-FINAL--Bench-black?style=for-the-badge" alt="FINAL-Bench"></a>
|
| 21 |
+
</p>
|
| 22 |
+
|
| 23 |
+
Darwin-180B-NL2SQL is VIDRAFT's text-to-SQL system built on [Darwin-180B-RSI](https://huggingface.co/FINAL-Bench/Darwin-180B-RSI). It turns a natural-language question into a SQLite query, runs it against the database and revises the query from what the execution shows, then returns one final query per question.
|
| 24 |
+
|
| 25 |
+
This repository holds the system's predictions on BIRD Mini-Dev (SQLite) and the script used to score them.
|
| 26 |
+
|
| 27 |
+
> **Weights:** the model is [Darwin-180B-RSI](https://huggingface.co/FINAL-Bench/Darwin-180B-RSI), unchanged. Download the weights from that repository; no weights were changed or trained for this result.
|
| 28 |
+
|
| 29 |
+
```python
|
| 30 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 31 |
+
|
| 32 |
+
repo = "FINAL-Bench/Darwin-180B-RSI" # weights for Darwin-180B-NL2SQL
|
| 33 |
+
tok = AutoTokenizer.from_pretrained(repo)
|
| 34 |
+
model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype="auto", device_map="auto")
|
| 35 |
+
```
|
| 36 |
+
|
| 37 |
+
## Result
|
| 38 |
+
|
| 39 |
+
**BIRD Mini-Dev, SQLite, 500 questions: 83.80% execution accuracy (419 / 500).**
|
| 40 |
+
|
| 41 |
+
| Difficulty | Correct | Accuracy |
|
| 42 |
+
|---|---|---|
|
| 43 |
+
| Challenging | 384 / 460 | 83.5% |
|
| 44 |
+
| Moderate | 29 / 32 | 90.6% |
|
| 45 |
+
| Simple | 6 / 8 | 75.0% |
|
| 46 |
+
| **All** | **419 / 500** | **83.80%** |
|
| 47 |
+
|
| 48 |
+
- Data: the current canonical release on Hugging Face, [`birdsql/bird_mini_dev`](https://huggingface.co/datasets/birdsql/bird_mini_dev), release "Mini-Dev 1024", `data/mini_dev_sqlite.jsonl` (sha256 `926af2a0d1b47e5d0d15a7aa8a34d4721007f4550a881555b6a89496b54a2606`), with the SQLite databases from `minidev.zip`.
|
| 49 |
+
- Metric: strict execution accuracy. A prediction is correct when the set of rows it returns equals the set of rows the reference SQL returns. Each predicted query may run for 60 seconds.
|
| 50 |
+
- Measured on 29 October 2026 (KST). The score was computed twice on an idle machine and was identical both times.
|
| 51 |
+
|
| 52 |
+
## Evaluation conditions
|
| 53 |
+
|
| 54 |
+
| Condition | Setting |
|
| 55 |
+
|---|---|
|
| 56 |
+
| Reference (gold) SQL used at any stage of inference | **No** |
|
| 57 |
+
| Oracle knowledge beyond the question and evidence given | No |
|
| 58 |
+
| Fine-tuning on BIRD | No |
|
| 59 |
+
| Final candidates per question | 1 |
|
| 60 |
+
| Database access during inference | Read-only execution of the system's own queries |
|
| 61 |
+
|
| 62 |
+
The question and evidence in each item are the only task inputs. The reference SQL was used only to score the final predictions.
|
| 63 |
+
|
| 64 |
+
## Files
|
| 65 |
+
|
| 66 |
+
| File | Contents |
|
| 67 |
+
|---|---|
|
| 68 |
+
| `predict_mini_dev_darwin-180b-nl2sql_SQLite.json` | Predictions in the BIRD Mini-Dev submission format (`SQL\t----- bird -----\tdb_id`, keyed by question ID) |
|
| 69 |
+
| `predictions.jsonl` | The same predictions, one JSON object per line |
|
| 70 |
+
| `score_ex.py` | Strict execution-accuracy scorer used for the numbers above |
|
| 71 |
+
|
| 72 |
+
To reproduce the score, place `mini_dev_sqlite.jsonl` next to `score_ex.py` and run:
|
| 73 |
+
|
| 74 |
+
```bash
|
| 75 |
+
python score_ex.py predictions.jsonl path/to/MINIDEV/dev_databases
|
| 76 |
+
```
|
| 77 |
+
|
| 78 |
+
## About the system
|
| 79 |
+
|
| 80 |
+
The system pairs Darwin-180B-RSI with VIDRAFT's NL2SQL verification engine, which checks each generated query against the live database before an answer is returned. The engine is VIDRAFT's own implementation and is not released.
|
| 81 |
+
|
| 82 |
+
A public demo of VIDRAFT's NL2SQL engine (Korean and English, read-only, synthetic data) is available at [final-bench-nl2sql-demo.hf.space](https://final-bench-nl2sql-demo.hf.space/).
|
| 83 |
+
|
| 84 |
+
## Contact
|
| 85 |
+
|
| 86 |
+
VIDRAFT 路 [vidraft.net](https://vidraft.net) 路 vidraft@mail.vidraft.net
|