open-reason-medium / README.md
theworker02's picture
Document 13.9M CPU medium model card
cf1c9f2 verified
|
Raw
History Blame Contribute Delete
1.58 kB
metadata
language:
  - en
license: apache-2.0
library_name: transformers
tags:
  - open-reason
  - causal-lm
  - cpu
datasets:
  - theworker02/open-reason
base_model: gpt2-scratch

Open Reason medium (CPU)

A medium GPT-2-style causal LM trained from scratch on theworker02/open-reason pipeline v1.4.0. It is larger than theworker02/open-reason-small (~1.3M) and is not a 1B model. Do not confuse it with theworker02/open-reason-1b.

Parameters 13,867,008
Architecture GPT-2 scratch, n_layer=6, n_embd=384, n_head=6, vocab 8192, context 192
Steps 180
Batch size 2
Hardware Host CPU (torch 2.12.0+cpu). Docker was not installed. AMD GPU was not used. CUDA: false
Dataset theworker02/open-reason v1.4.0, 3175 SFT rows (all split)
Final loss 4.416
License Apache-2.0
Reddit Never used as a source
from transformers import AutoModelForCausalLM, AutoTokenizer

tok = AutoTokenizer.from_pretrained("theworker02/open-reason-medium")
model = AutoModelForCausalLM.from_pretrained("theworker02/open-reason-medium")

Companion small model: theworker02/open-reason-small. Dataset: theworker02/open-reason. Code: theworker02/open-reason.