Skip to content

The training code to finetune flan-T5 model with text to SQL task

License

Notifications You must be signed in to change notification settings

juierror/flan-t5-text-to-sql

Repository files navigation

Flan T5 Text to SQL

This repository contains code for finetuning the Flan T5 model for the text-to-SQL task. The text-to-SQL task is the problem of mapping natural language questions to SQL queries that can be executed on a database.

Environment Setup

pip3 install -r requirements.txt

# for the pytorch installation, it is depends on which version you should use (linux, mac or have a CUDA or not). You should check https://pytorch.org

Training

Please run through the flan-t5-text2sql.ipynb code.

Reference

About

The training code to finetune flan-T5 model with text to SQL task

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published