BRIO 모델

안민기·2023년 3월 3일
0

paper with code에서 Abstractive Text Summarization on CNN / Daily Mail Benchmark중 23년 3월 현재 3번째로 랭크되있는 모델
페가수스모델에 기반을 하고 있음

모델 사용 방법

1. git clone : https://github.com/yixinl7/brio#use-brio-with-huggingface

2. Huggingface :

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')

모델의 VRAM 확인하기


2129MB => 2.23GB
생각보다 낮은 VRAM

USE

당연히 한글 지원은 안됨

원문 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.

profile
Trendy AI Developer

0개의 댓글