How to use Ahmed167/pegasus-dialogue with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("Ahmed167/pegasus-dialogue") model = AutoModelForSeq2SeqLM.from_pretrained("Ahmed167/pegasus-dialogue", device_map="auto")
