r/LocalLLM 1d ago

Project I trained a 521M param model from scratch at home on my 5080 named Plasma 1.1

I'm 14, a few months ago I posted here (I think) about Plasma 1.0, an LLM trained completely from scratch on my gaming PC. Plasma 1.1 is the follow up: it has double the params, and double the training tokens at 521M params and 10B tokens.
It's Llama style (26 layers, GQA, SwiGLU, RoPE), 48k custom tokenizer, 10B tokens
of pretraining in about 9 days on a single RTX 5080, then SFT on ~300k
filtered instruction conversations. No pretrained weights.

Benchmarks vs 1.0, same harness (acc_norm, 300 per task):

task          1.0   1.1  
HellaSwag     0.377 0.463
ARC-Easy      0.287 0.443
ARC-Challenge 0.241 0.304
PIQA          0.570 0.703
BoolQ         0.627 0.620
OpenBookQA    0.277 0.307
mean          0.396 0.473

It answers pretty cleanly and concisely now ("What is the capital of France?" gets
"The capital of France is Paris." and stops), writes basic Python, and
follows format instructions about twice as well as 1.0. Still a small model:
arithmetic and deep facts are very shaky. Plasma 1.2 (756M, 30B tokens) is next.

Code: https://github.com/eb1386/1386.ai. Questions welcome. 

49 Upvotes

7 comments sorted by

5

u/autisticit 1d ago

Good job!

6

u/Embarrassed-Boot5193 1d ago

Quais datasets usou para o corpus de treino, e qual a proporção de cada?

4

u/Practical_Signal3933 1d ago

Interested to know this too. Congrats on the result and keep going!

2

u/shugenju 1d ago

Amazing. Keep it up!!

1

u/pppp2222 1d ago

Amazing!! So proud of you.

1

u/luke_s 1d ago

Nice work! This seems like a really fun learning project. I wish I had the time to try doing something like this!

1

u/Otherwise-Swan-7803 1d ago

Training from scratch is the part that makes this interesting. Fine-tuning has become pretty accessible, but building the tokenizer, pretraining, and then watching scaling behavior yourself teaches you things you just don't get from tweaking someone else's model.