Update README.md
Browse files
README.md
CHANGED
|
@@ -6,6 +6,9 @@ tags:
|
|
| 6 |
- pytorch
|
| 7 |
- rnn
|
| 8 |
- text-generation
|
|
|
|
|
|
|
|
|
|
| 9 |
datasets:
|
| 10 |
- aditya-6122/tinystories-custom-dataset-18542-v2-test
|
| 11 |
pipeline_tag: text-generation
|
|
@@ -19,8 +22,7 @@ pipeline_tag: text-generation
|
|
| 19 |
This is a custom language model trained on a dataset of short stories, designed for text generation tasks.
|
| 20 |
|
| 21 |
### Model Sources
|
| 22 |
-
- **Repository**: [
|
| 23 |
-
- **Paper**: N/A
|
| 24 |
|
| 25 |
## Uses
|
| 26 |
|
|
@@ -81,7 +83,8 @@ model.eval()
|
|
| 81 |
|
| 82 |
# Generate text
|
| 83 |
input_text = "Once upon a time"
|
| 84 |
-
|
|
|
|
| 85 |
```
|
| 86 |
|
| 87 |
## Limitations
|
|
|
|
| 6 |
- pytorch
|
| 7 |
- rnn
|
| 8 |
- text-generation
|
| 9 |
+
- gru
|
| 10 |
+
- tiny-stories
|
| 11 |
+
- bpe-tokenizer
|
| 12 |
datasets:
|
| 13 |
- aditya-6122/tinystories-custom-dataset-18542-v2-test
|
| 14 |
pipeline_tag: text-generation
|
|
|
|
| 22 |
This is a custom language model trained on a dataset of short stories, designed for text generation tasks.
|
| 23 |
|
| 24 |
### Model Sources
|
| 25 |
+
- **Repository**: [Aditya6122/BuildingLanguageModel-TinyStories](https://github.com/Aditya6122/BuildingLanguageModel-TinyStories)
|
|
|
|
| 26 |
|
| 27 |
## Uses
|
| 28 |
|
|
|
|
| 83 |
|
| 84 |
# Generate text
|
| 85 |
input_text = "Once upon a time"
|
| 86 |
+
|
| 87 |
+
# Tokenize and generate [Add your Generation Logic]
|
| 88 |
```
|
| 89 |
|
| 90 |
## Limitations
|