Downloads · 30 days
7
11% of all-time downloads
AbhijithSK/mistral-7b-medical-instruct
mistral-7b-medical-instruct is a machine learning model from AbhijithSK. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft. The card lists the license as apache-2.0.
Fine-tuned version of mistralai/Mistral-7B-Instruct-v0.2 on medical Q&A data using QLoRA.
Downloads · 30 days
7
11% of all-time downloads
All-time downloads
64
Public
Repo size
301 MB
Likes
0
Public
Click a slice to open those files.
.pt164 MB · 52%
From the Hugging Face model README
Fine-tuned version of mistralai/Mistral-7B-Instruct-v0.2 on medical Q&A data using QLoRA.
| Metric | Value |
|---|---|
| Final Train Loss | 0.022 |
| Token Accuracy | 98.9% |
| Epochs | 3 |
| Training Time | ~2h 50min on T4 GPU |
| Trainable Params | 13.6M / 7.25B (0.19%) |
from peft import PeftModel from transformers import AutoModelForCausalLM, AutoTokenizer
base = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2") model = PeftModel.from_pretrained(base, "AbhijithSK/mistral-7b-medical-instruct") tokenizer = AutoTokenizer.from_pretrained("AbhijithSK/mistral-7b-medical-instruct")
For research only. Not for clinical use. Always consult a qualified physician.
Abhijith SK - GitHub: https://github.com/AbhijithSK