KookyGhost/txt2sql-LoRA-merged-t5-small

Model Card for Model ID

Convert natural language queries into SQL.

Model Details

Model Description

A text-to-sql LoRA adapter for m5-small finetuned on the b-mc2/sql-create-context, Clinton/Text-to-sql-v1, and knowrohit07/know_sql datasets.

• Model type: text-to-sql
• Language(s) (NLP): Python (PyTorch)
• Finetuned from model: m5-small

Model Sources [optional]

• Repository: github: Text2sql-LoRA-T5-small

Training Details

Training Data

b-mc2/sql-create-context

Clinton/Text-to-sql-v1

knowrohit07/know_sql

Train-Eval-Test split: 8:1:1

Metrics

ROUGE

Results

'rouge1': 0.9281907395888822, 'rouge2': 0.8917983416395394, 'rougeL': 0.9182615264948171, 'rougeLsum': 0.9191261027066355

Summary

This adapter helps m5-small achieve a ROUGE score of over 90% on text to SQL tasks.

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

• Hardware Type: RTX 4000

• Hours used: 2

• Cloud Provider: Paperspace

• PEFT 0.6.2