Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("ALI-USER/rl-grpo-sql-model")
4model = AutoModelForCausalLM.from_pretrained("ALI-USER/rl-grpo-sql-model")
5
6# Example usage
7prompt = "Generate SQL for: Find all customers with orders over $100"
8inputs = tokenizer(prompt, return_tensors="pt")
9outputs = model.generate(**inputs, max_length=512)
10print(tokenizer.decode(outputs[0]))