aditya-6122 commited on
Commit
0474187
·
verified ·
1 Parent(s): 06b9882

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -3
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**: [GitHub Repository](https://github.com/your-repo) # Replace with actual repo if available
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
- # Tokenize and generate (implement your generation logic)
 
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