Views
No views yet
| Property | Value |
|---|---|
| Parameters | ~40M |
| Layers | 12 |
| Hidden size | 512 |
| Attention heads | 8 |
| Context length | 256 tokens |
| Positional encoding | Rotary (RoPE) |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("username/ptbr-40m-base")
4tokenizer = AutoTokenizer.from_pretrained("gpt2")
5
6inputs = tokenizer("Olá mundo", return_tensors="pt")
7
8outputs = model(**inputs)