Update README.md

README.md

@@ -6,17 +6,31 @@ tags:
 - text-generation-inference
 - transformers
 - unsloth
-- llama
+- mistral
 - trl
+- medical
 base_model: unsloth/llama-2-7b-chat-bnb-4bit
+datasets:
+- Shekswess/medical_llama_instruct_dataset_short
 ---
 
-# Uploaded  model
-
 - **Developed by:** Shekswess
 - **License:** apache-2.0
 - **Finetuned from model :** unsloth/llama-2-7b-chat-bnb-4bit
 
-This llama model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
+To utilize the finetuning of the model, you need to use the gemma instruction prompt template for this medical version of the model:
+```
+[INST] Answer the question truthfully, you are a medical professional. This is the question: {question} [/INST]
+```
+
+Metrics: 
+
+- train_runtime: 4161.8304
+- train_samples_per_second: 0.721
+- train_steps_per_second: 0.09
+- total_flos: 5.622915296673792e+16
+- train_loss: 1.0950924614270527
+- steps: 375
+- epoch: 1.0
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6569f13004643352df96e40f/x4e5tNog69QDN0l2-9F8S.png)