Self hosted AI to train on an existing database

I have a SQL database, and a schema document that goes with it, is there any kind of AI I can self host and train it on this data?

The goal would be to ask it simple questions like “from which table in [dbname] can I find a list of products I’ve sold in the last x days”

Would be even better if I could ask it to write some queries to find exactly what I’m after.

What kind of hardware would I need to run something like this, if it exists?

You're viewing a single thread.

2 comments

You could use something like https://huggingface.co/defog/sqlcoder-34b-alpha / https://github.com/defog-ai/sqlcoder however I haven’t used this one myself but it builds up on CodeLlama model so the quality should be good. I haven’t seen other models specifically for SQL queries yet.

Edit: it includes all the hardware requirements and some demo examples in the links