Update README.md

README.md

@@ -8,9 +8,7 @@ tags:
 datasets: pt-sk/research_papers_short
 metrics: CrossEntropyLoss
 ---
-This model using Mamba Architecuture trained on a research abstract dataset. 
-
-mamba_reserach file contains the state dict of optimizer and the model.
+This model uses Mamba Architecture trained on a research abstract dataset. 
 
 * Optimizer: AdamW
 * Leanring Rate: 0.001
@@ -31,4 +29,7 @@ mamba_model = Mamba.from_pretrained("pt-sk/mamba").to("cuda")
 
 Loading Tokenizer
 ```
-tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')
+tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')
+```
+
+mamba_reserach file contains the state dict of optimizer and the model.