Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,9 @@ pipeline_tag: text-generation
|
|
| 8 |
---
|
| 9 |
|
| 10 |
We tried to use the huggingface transformers library to recreate the TinyStories models on Consumer GPU. Output model has 9 million parameters.
|
|
|
|
| 11 |
Tweaked code of springtangent (https://github.com/springtangent/tinystoriestrainer/blob/main/tinystories_train.py)
|
|
|
|
| 12 |
Code credit - springtangent
|
| 13 |
|
| 14 |
------ EXAMPLE USAGE ---
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
We tried to use the huggingface transformers library to recreate the TinyStories models on Consumer GPU. Output model has 9 million parameters.
|
| 11 |
+
|
| 12 |
Tweaked code of springtangent (https://github.com/springtangent/tinystoriestrainer/blob/main/tinystories_train.py)
|
| 13 |
+
|
| 14 |
Code credit - springtangent
|
| 15 |
|
| 16 |
------ EXAMPLE USAGE ---
|