from peft import PeftModel, PeftConfig
from transformers import AutoModelForCausalLM

config = PeftConfig.from_pretrained("therealcyberlord/TinyLlama-1.1B-Medical")
model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
model = PeftModel.from_pretrained(model, "therealcyberlord/TinyLlama-1.1B-Medical")

Training Details

Training Data

Used two data sources:

BI55/MedText: https://huggingface.co/datasets/BI55/MedText

MedQuad-MedicalQnADataset: https://huggingface.co/datasets/keivalya/MedQuad-MedicalQnADataset

Training Procedure

Trained on 1000 steps on a shuffled combined dataset

Framework versions

PEFT 0.7.2.dev0