Update README.md

README.md

@@ -6,28 +6,24 @@ tags:
 - Test Generation
 - research abstract
 datasets: pt-sk/research_papers_short
+metrics: CrossEntropyLoss
 ---
-```
-from huggingface_hub import login
-login()
 
-from huggingface_hub import HfApi
-api = HfApi()
-api.upload_file(
-    path_or_fileobj="/content/drive/MyDrive/mamba_research_test",
-    path_in_repo="mamba_research_test",
-    repo_id="pt-sk/mamba_scratch",
-    repo_type="model",
-    commit_message="uploaded model file"
-)
+Optimizer: AdamW
+Leanring Rate: 0.001
+
 
-from huggingface_hub import HfApi
-api = HfApi()
 
-api.upload_folder(
-    folder_path="/content/tokenizer",
-    repo_id="pt-sk/mamba_scratch",
-    repo_type="model",
+Import the scripts from the code folder
+```
+from model import Mamba, ModelArgs
+```
+
+Loading Model
+```
+mamba_model = Mamba.from_pretrained("pt-sk/mamba").to("cuda")
+```
 
-)
-```
+Loading Tokenizer
+```
+tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')