Views
No views yet
SQL-Gemma3 is a fine-tuned version of Gemma 3 1B Instruct for text-to-SQL generation. It was trained on a balanced sampled subset of the Gretel synthetic_text_to_sql dataset to improve SQL generation from table schema and natural language questions.unsloth/gemma-3-1b-itgretelai/synthetic_text_to_sql0.2010.211from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_id = "vishnurchityala/sql-gemma3"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8messages = [
9 {
10 "role": "user",
11 "content": (
12 "CREATE TABLE employees(id INT, name TEXT, salary INT);\n\n"
13 "Find the average salary of all employees."
14 ),
15 }
16]
17
18inputs = tokenizer(
19 tokenizer.apply_chat_template(
20 messages,
21 tokenize=False,
22 add_generation_prompt=True,
23 ),
24 return_tensors="pt",
25)
26
27outputs = model.generate(**inputs, max_new_tokens=128, do_sample=False)
28print(tokenizer.decode(outputs[0], skip_special_tokens=True))