File size: 1,583 Bytes
2e526ba
 
 
 
 
 
 
 
 
 
 
 
 
 
cf1c9f2
2e526ba
cf1c9f2
 
 
 
 
2e526ba
cf1c9f2
 
 
 
 
 
 
 
 
 
 
2e526ba
 
 
cf1c9f2
2e526ba
 
 
cf1c9f2
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
---

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`](https://huggingface.co/datasets/theworker02/open-reason)
pipeline **v1.4.0**. It is larger than
[`theworker02/open-reason-small`](https://huggingface.co/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 |

```python

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`](https://huggingface.co/theworker02/open-reason-small).
Dataset: [`theworker02/open-reason`](https://huggingface.co/datasets/theworker02/open-reason).
Code: [`theworker02/open-reason`](https://github.com/theworker02/open-reason).