Downloads · 30 days
3
4% of all-time downloads
Dodosoomro/simple-100m-pretrain-v1
simple-100m-pretrain-v1 is a machine learning model from Dodosoomro. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This is a 120M parameter decoder-only transformer trained from scratch on 1 billion tokens of high-quality web data, math, and code.
Downloads · 30 days
3
4% of all-time downloads
All-time downloads
76
Public
Parameters
146M
292 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors292 MB · 100%
From the Hugging Face model README
This is a 120M parameter decoder-only transformer trained from scratch on 1 billion tokens of high-quality web data, math, and code.
| Metric | Task | Value |
|---|---|---|
| Perplexity | Training-Validation | 35.04 |
| Accuracy | ARC-Easy | 31.10% |
| Accuracy | PIQA | 54.13% |
| Accuracy | HellaSwag | 26.22% |
| Accuracy | TruthfulQA (MC1) | 27.42% |
| Accuracy | GSM8K (5-shot) | 1.36% |
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("Dodosoomro/simple-100m-pretrain-v1")
tokenizer = AutoTokenizer.from_pretrained("Dodosoomro/simple-100m-pretrain-v1")
prompt = "The future of artificial intelligence is"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=50)
print(tokenizer.decode(outputs[0]))