llama3.2-3B-text-to-sql

Llama 3.2 3B — Text-to-SQL

A fine-tuned version of Meta Llama 3.2 3B for converting natural-language questions into SQL queries.

The model was fine-tuned using QLoRA with a 4-bit quantized base model and LoRA adapters. After training, the adapter was merged with the base model to produce this standalone model.

Model Repositories

Merged Model

The base model and trained LoRA adapter have been merged into a standalone model:

https://huggingface.co/farehaheha/llama3.2-3B-text-to-sql

Quantized GGUF Model

A Q4_K_M GGUF quantized version is available for efficient local inference with llama.cpp and other GGUF-compatible runtimes:

https://huggingface.co/farehaheha/llama3.2-3B-text-to-sql-Q4_K_M-GGUF

Training Details

Note: This experiment used the dataset's test split for fine-tuning rather than the training split. Therefore, the original test split should not be used to report an unbiased evaluation score for this model.

Dataset

The model was fine-tuned on:

Gretel AI Synthetic Text-to-SQL

https://huggingface.co/datasets/gretelai/synthetic_text_to_sql

The dataset contains natural-language questions paired with SQL queries and database context.

Approximately 5,850 examples from the test split were used for this fine-tuning experiment.

Fine-Tuning Approach

The training process used QLoRA:

1. The Llama 3.2 3B base model was loaded in 4-bit precision.
2. The original base model weights remained frozen.
3. LoRA adapters with rank 16 were trained on the Text-to-SQL dataset.
4. Training was performed for one epoch.
5. The trained LoRA adapter was merged with the base model.
6. The merged model was also converted to Q4_K_M GGUF for local inference.

This approach significantly reduces the memory required for fine-tuning compared with full-parameter training.

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM

model_id = "farehaheha/llama3.2-3B-text-to-sql"

tokenizer = AutoTokenizer.from_pretrained(model_id)

model = AutoModelForCausalLM.from_pretrained(
    model_id,
    device_map="auto",
    torch_dtype="auto",
)

prompt = """### Database Schema:
{your_database_schema}

### Request:
{your_natural_language_request}

### SQL Query:
"""

inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

outputs = model.generate(
    **inputs,
    max_new_tokens=256,
    do_sample=False,
)

response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)