Update README.md

README.md

@@ -6,17 +6,31 @@ tags:
 - text-generation-inference
 - transformers
 - unsloth
-- gemma
+- mistral
 - trl
+- medical
 base_model: unsloth/gemma-1.1-7b-it-bnb-4bit
+datasets:
+- Shekswess/medical_gemma_instruct_dataset_short
 ---
 
-# Uploaded  model
-
 - **Developed by:** Shekswess
 - **License:** apache-2.0
 - **Finetuned from model :** unsloth/gemma-1.1-7b-it-bnb-4bit
 
-This gemma 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 :
+```
+<start_of_turn>user Answer the question truthfully, you are a medical professional. This is the question: {question}<end_of_turn>
+```
+
+Metrics: 
+
+- train_runtime: 4544.6901
+- train_samples_per_second: 0.66
+- train_steps_per_second: 0.083
+- total_flos: 5.862681923842867e+16
+- train_loss: 1.7245690104166667
+- 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/L22u9s2ShJRwg3AdYs-06.png)