knkarthick/dialogsum
Viewer • Updated • 14.5k • 7.13k • 246
How to use Mia2024/CS5100TextSummarization with Transformers:
# 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.