paper with code에서 Abstractive Text Summarization on CNN / Daily Mail Benchmark중 23년 3월 현재 3번째로 랭크되있는 모델
페가수스모델에 기반을 하고 있음
from transformers import BartTokenizer, PegasusTokenizer
from transformers import BartForConditionalGeneration, PegasusForConditionalGeneration
model = PegasusForConditionalGeneration.from_pretrained('Yale-LILY/brio-xsum-cased')
tokenizer = PegasusTokenizer.from_pretrained('Yale-LILY/brio-xsum-cased')
2129MB => 2.23GB
생각보다 낮은 VRAM
당연히 한글 지원은 안됨
원문 1
"Manchester United superstar Cristiano Ronaldo scored his 806th career goal in Old Trafford,\
breaking FIFA's all-time record for most goals in competitive matches in men's football history.\
It was the second of three goals the Portuguese attacker scored during the game,\
leading United to a 3-2 victory over Tottenham and finishing the day with 807 total career goals.\
The previous FIFA goal record was held by Josef Bican, with 805 goals."
요약 1
Cristiano Ronaldo scored his 806th career goal in a 3-2 win over Tottenham at Old Trafford. Ronaldo breaks FIFA's record for most goals in competitive matches in men's football history. The Portuguese scored the second of three goals to reach 807 career goals. The previous record was held by Josef Bican.
원문 2
"On 24 February 2022, Russia launched a military invasion of Ukraine in a steep escalation of the Russo-Ukrainian War.\
The campaign had been preceded by a Russian military buildup since early 2021 and numerous Russian demands for security measures and legal prohibitions against Ukraine joining NATO."
요약 2
Russia launched a military invasion of Ukraine on 24 February 2022. the invasion was a steep escalation of the Russo-Ukrainian War. the campaign had been preceded by a Russian military buildup since early 2021. Russian demands for Ukraine joining NATO were included in the invasion.