File size: 2,366 Bytes
f2bd10f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 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 | ---
library_name: onnx
tags:
- text2text-generation
- t5
- text-to-sql
- sql
- spider
- encoder-decoder
- onnx
- inference4j
license: apache-2.0
pipeline_tag: text2text-generation
datasets:
- spider
- spider-syn
---
# T5-LM-Large text2sql-spider — ONNX
ONNX export of [T5-LM-Large-text2sql-spider](https://huggingface.co/gaussalgo/T5-LM-Large-text2sql-spider) (0.8B parameters) with encoder-decoder architecture and KV cache support.
This is a T5-large model fine-tuned on the Spider and Spider-Syn datasets for text-to-SQL generation. Given a natural language question and a database schema, it produces the corresponding SQL query.
Converted for use with [inference4j](https://github.com/inference4j/inference4j), an inference-only AI library for Java.
## Original Source
- **Repository:** [gaussalgo/T5-LM-Large-text2sql-spider](https://huggingface.co/gaussalgo/T5-LM-Large-text2sql-spider)
- **Base model:** [google/t5-large-lm-adapt](https://huggingface.co/google/t5-large-lm-adapt)
- **License:** Apache 2.0
## Usage with inference4j
```java
try (var sqlGen = T5SqlGenerator.t5LargeSpider().build()) {
String sql = sqlGen.generateSql(
"How many employees are in each department?",
"\"employees\" \"id\" int, \"name\" varchar, \"dept_id\" int "
+ "[SEP] \"departments\" \"id\" int, \"name\" varchar");
System.out.println(sql);
}
```
## Schema Format
The model expects the schema in the following format:
```
"table_name" "col1" type, "col2" type, foreign_key: "table"."col" = "other"."col" primary key: "col" [SEP] "table2" ...
```
- Table and column names are double-quoted
- Columns are comma-separated with types
- Tables are separated by `[SEP]`
- Foreign keys and primary keys are declared per table
## Model Details
| Property | Value |
|----------|-------|
| Architecture | T5 encoder-decoder (0.8B parameters) |
| Task | Text-to-SQL generation |
| Training data | Spider, Spider-Syn |
| Tokenizer | SentencePiece (32,128 tokens) |
| Original framework | PyTorch (transformers) |
| Export method | Hugging Face Optimum (encoder-decoder with KV cache) |
## License
This model is licensed under the [Apache License 2.0](https://www.apache.org/licenses/LICENSE-2.0). Original model by [Gaussalgo](https://huggingface.co/gaussalgo), base model by [Google](https://huggingface.co/google).
|