Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("builderbrain_tiny_1759367360")
4model = AutoModelForCausalLM.from_pretrained("builderbrain_tiny_1759367360")
5
6# Grammar-constrained generation
7input_text = "Generate a JSON API call for user registration"
8inputs = tokenizer(input_text, return_tensors="pt")
9
10# Generate with grammar constraints and safety monitoring
11outputs = model.generate(
12 **inputs,
13 max_length=150,
14 grammar_constraint=True,
15 safety_monitoring=True,
16 temperature=0.8
17)
18response = tokenizer.decode(outputs[0], skip_special_tokens=True)1@misc{builderbrain_tiny,
2 title={BuilderBrain: Dual-Rail Compositional AI System},
3 author={BuilderBrain Team},
4 year={2024},
5 url={https://github.com/JacobFV/builderbrain}
6}