Update README.md

README.md

@@ -8,10 +8,14 @@ tags:
 - mixture-of-experts
 - autoregressive
 - text-generation
+- 73.8 M
+- pretrained
 datasets:
 - HuggingFaceFW/fineweb-edu
 pipeline_tag: text-generation
 library_name: pytorch
+model_type: custom
+license: mit
 ---
 
 # HSSM
@@ -185,4 +189,4 @@ print(reply)
 
 ## Citation
 
-If you use this release in experiments, please cite the model repository and mention the FineWeb-Edu pretraining source.
+If you use this release in experiments, please cite the model repository and mention the FineWeb-Edu pretraining source.