bart1259 commited on
Commit
182cf00
·
verified ·
1 Parent(s): f5f3af2

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +12 -11
  2. model.safetensors +1 -1
  3. score.png +0 -0
README.md CHANGED
@@ -10,8 +10,11 @@ Model details:
10
  - **Attention Head Count**: 4
11
  - **Residual Stream Size**: 256
12
  - **Context Length**: 256
13
- - **Tokens Trained on**: 419,706,880
14
 
 
 
 
15
 
16
  ```py
17
  from transformers import pipeline
@@ -19,7 +22,8 @@ from transformers import pipeline
19
  pipe = pipeline(
20
  "text-generation", model="bart1259/MiniCOTMath"
21
  )
22
- print(pipe("Input: (5 + 5)\n", max_new_tokens=100)[0]["generated_text"])
 
23
  ```
24
 
25
  Outputs:
@@ -59,7 +63,8 @@ class StopCriteria(StoppingCriteria):
59
  def __iter__(self):
60
  yield self
61
 
62
- prompt = "Input: (3 * (5 + 5))\n"
 
63
 
64
  tokenizer = AutoTokenizer.from_pretrained("bart1259/MiniCOTMath")
65
  model = AutoModelForCausalLM.from_pretrained("bart1259/MiniCOTMath").cuda()
@@ -81,18 +86,14 @@ _ = model.generate(
81
 
82
  Outputs:
83
  ```
84
- Input: (3 * (5 + 5))
85
 
86
  Step 1:
87
- (3 * (5 + 5))
88
  (5 + 5) = 10
89
 
90
  Step 2:
91
- (3 * 10)
92
- (3 * 10) = 30
93
-
94
- Step 3:
95
- 30
96
- Final Result: 30
97
  <end>
98
  ```
 
10
  - **Attention Head Count**: 4
11
  - **Residual Stream Size**: 256
12
  - **Context Length**: 256
13
+ - **Tokens Trained on**: 419,649,024
14
 
15
+ Training Score During Training
16
+
17
+ [score.png](score.png)
18
 
19
  ```py
20
  from transformers import pipeline
 
22
  pipe = pipeline(
23
  "text-generation", model="bart1259/MiniCOTMath"
24
  )
25
+ print(pipe("Input: (5 + 5)
26
+ ", max_new_tokens=100)[0]["generated_text"])
27
  ```
28
 
29
  Outputs:
 
63
  def __iter__(self):
64
  yield self
65
 
66
+ prompt = "Input: (5 + 5)
67
+ "
68
 
69
  tokenizer = AutoTokenizer.from_pretrained("bart1259/MiniCOTMath")
70
  model = AutoModelForCausalLM.from_pretrained("bart1259/MiniCOTMath").cuda()
 
86
 
87
  Outputs:
88
  ```
89
+ Input: (5 + 5)
90
 
91
  Step 1:
92
+ (5 + 5)
93
  (5 + 5) = 10
94
 
95
  Step 2:
96
+ 10
97
+ Final Result: 10
 
 
 
 
98
  <end>
99
  ```
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b0e7b2d91eb4e7ea8bdc4b94640382a0af079b8e8e9974b801c9c48fdf4af330
3
  size 25352072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80470ec0e1be3e0e2ae474707fdd150d02d666915dfaa5d54528418c03e7da66
3
  size 25352072
score.png ADDED