Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model and tokenizer
4model = AutoModelForCausalLM.from_pretrained("exaler/aaa-2-sql")
5tokenizer = AutoTokenizer.from_pretrained("exaler/aaa-2-sql")
6
7# Create prompt
8prompt = "Your prompt here"
9
10# Generate text
11inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
12output = model.generate(**inputs, max_new_tokens=1024)
13response = tokenizer.decode(output[0], skip_special_tokens=True)
14print(response)