knkarthick/dialogsum
Viewer β’ Updated β’ 14.5k β’ 7.08k β’ 259
How to use Mia2024/CS5100TextSummarization with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("Mia2024/CS5100TextSummarization")
model = AutoModelForSeq2SeqLM.from_pretrained("Mia2024/CS5100TextSummarization", device_map="auto")Final Project for 2024 Fall CS5100 Foundations of Artificial Intelligence at Roux Institute, Northeastern University.
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("Mia2024/CS5100TextSummarization") model = AutoModelForSeq2SeqLM.from_pretrained("Mia2024/CS5100TextSummarization", device_map="auto")