Views
No views yet
transformers library for training and finetuning.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("efederici/ipt-125m", trust_remote_code=True)
4tokenizer = AutoTokenizer.from_pretrained("efederici/ipt-125m")| Hyperparameter | Value |
|---|---|
| n_parameters | 125M |
| n_layers | 12 |
| n_heads | 12 |
| d_model | 768 |
| vocab size | 50432 |
| sequence length | 2048 |