Views
No views yet
| Epoch | Dataset Size | Learning Rate | Focus |
|---|---|---|---|
| 1 | 36K | 2e-4 | Core instruction following |
| 2 | 36K | 5e-5 | Continued refinement |
| 3 | 98K | 2e-5 | Expanded: safety, Indic languages, clean instructions |
| 4 | 98K | 1e-5 | Careful refinement (low LR, anti-forgetting) |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("laabamone/laabam-ai-3b-v1")
4tokenizer = AutoTokenizer.from_pretrained("laabamone/laabam-ai-3b-v1")
5
6messages = [{"role": "user", "content": "Hello, who are you?"}]
7inputs = tokenizer.apply_chat_template(messages, return_tensors="pt")
8outputs = model.generate(inputs, max_new_tokens=256)
9print(tokenizer.decode(outputs[0], skip_special_tokens=True))