Views
No views yet
| Parameter | Value |
|---|---|
| Hidden Size | 2048 |
| Layers | 24 |
| Attention Heads | 16 |
| KV Heads | 4 (GQA) |
| Intermediate Size | 5504 |
| Vocab Size | 32000 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("limajr/nbr-1b-base")
4tokenizer = AutoTokenizer.from_pretrained("limajr/nbr-1b-base")
5
6text = "O Brasil e um pais"
7inputs = tokenizer(text, return_tensors="pt")
8outputs = model.generate(**inputs, max_new_tokens=100)
9print(tokenizer.decode(outputs[0]))