How to use Ketan3101/ConvoBrief with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn") model = PeftModel.from_pretrained(base_model, "Ketan3101/ConvoBrief")
How to use Ketan3101/ConvoBrief with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("summarization", model="Ketan3101/ConvoBrief")
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Ketan3101/ConvoBrief", dtype="auto")
librarian-bot