lavita/MedQuAD
Viewer • Updated • 47.4k • 3.26k • 22
How to use vlachner/mistral-medquad-r32 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-r32")Training configuration
| Step | Training Loss | Validation Loss | Mean Token Accuracy |
|---|---|---|---|
| 20 | 1.1077 | 1.0624 | 0.7329 |
| 100 | 0.7190 | 0.8991 | 0.7721 |
| 200 | 0.9566 | 0.8750 | 0.7758 |
| 300 | 0.9045 | 0.8575 | 0.7783 |
| 400 | 1.0345 | 0.8418 | 0.7806 |
| 500 | 0.9336 | 0.8355 | 0.7817 |
Final metrics
Model fine-tuned on the MedQuAD dataset for medical QA using PEFT + QLoRA with rank = 32.
Base model
mistralai/Mistral-7B-v0.3