FlowerTune Lora Collection
Collection
12 items • Updated
How to use zjudai/flowertune-medical-lora-mistral-7b-instruct-v0.3 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, "zjudai/flowertune-medical-lora-mistral-7b-instruct-v0.3")This is a LoRA adapter for mistralai/Mistral-7B-Instruct-v0.3 fine-tuned with Flower federated learning framework on medical NLP datasets.
This model is a LoRA adapter fine-tuned on mistralai/Mistral-7B-Instruct-v0.3 using the Flower federated learning framework. It was trained on medical NLP datasets through distributed learning. The model achieved scores of PubMedQA: 0.5440, MedMCQA: 0.5520, MedQA: 0.5440, CareQA: 0.4980, with an average score of 0.5345.
Base model
mistralai/Mistral-7B-v0.3