lavita/MedQuAD
Viewer • Updated • 47.4k • 2.98k • 22
How to use vlachner/mistral-medquad-r4 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
model = PeftModel.from_pretrained(base_model, "vlachner/mistral-medquad-r4")Training configuration
| Step | Training Loss | Validation Loss | Mean Token Accuracy |
|---|---|---|---|
| 20 | 1.1792 | 1.1783 | 0.7006 |
| 100 | 0.7425 | 0.9143 | 0.7697 |
| 200 | 0.9692 | 0.8869 | 0.7738 |
| 300 | 0.9178 | 0.8768 | 0.7758 |
| 400 | 1.0642 | 0.8701 | 0.7773 |
| 500 | 0.9679 | 0.8673 | 0.7783 |
Final metrics
Model fine-tuned on the MedQuAD dataset for medical QA using PEFT + QLoRA.
Base model
mistralai/Mistral-7B-v0.3