DistilBART SAMSum LoRA
This repository contains a LoRA adapter fine-tuned on the SAMSum dataset for dialogue summarization.
Base Model
sshleifer/distilbart-cnn-12-6
Usage
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
from peft import PeftModel
base = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
model = PeftModel.from_pretrained(base, "username/distilbart-samsum-lora")
- Downloads last month
- 5