Views
No views yet

1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tok = AutoTokenizer.from_pretrained("KSP-NMAI/Boris-125M")
4model = AutoModelForCausalLM.from_pretrained("KSP-NMAI/Boris-125M")
5
6ids = tok("The ocean is", return_tensors="pt").input_ids
7out = model.generate(ids, max_new_tokens=40, do_sample=True, top_p=0.95)
8print(tok.decode(out[0], skip_special_tokens=True))| Architecture | GPT-2 (pre-LN, learned positional embeddings, tied embeddings) |
| Layers / heads / d_model | 12 / 12 / 768 |
| Context length | 1024 |
| Vocab | 50304 (GPT-NeoX-20B BPE, padded) |
| Tokenizer | EleutherAI/gpt-neox-20b |
| Precision | trained in bf16 autocast with fp32 master weights |
| Final loss | 3.2998 |
| Final grad norm | 0.281 |
| Final learning rate | 6.00e-05 |

http://www.apache.org/licenses/LICENSE-2.0