KoBART_with_LED / README.md
noahkim's picture
Update README.md
ad12c6b
---
language: ko
tags:
- summarization
- bart
- longformer
---
- This model is a [kobart-news](https://huggingface.co/ainize/kobart-news) finetuned on the [๋ฌธ์„œ์š”์•ฝ ํ…์ŠคํŠธ/์‹ ๋ฌธ๊ธฐ์‚ฌ](https://aihub.or.kr/aidata/8054)
<<20220904 Commit>>
๊ฐœ์ธ ์Šคํ„ฐ๋””์šฉ์œผ๋กœ ๊ธด ๋ฌธ์žฅ์˜ ์š”์•ฝ ๋ชจ๋ธ ํŠนํ™”๋œ ๋ชจ๋ธ์„ ๋งŒ๋“ค๊ธฐ ์œ„ํ•ด KoBART ๋ชจ๋ธ์„ Longformer Encoder Decoder๋กœ ๋ณ€ํ™˜ํ•œ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.
๊ธฐ์กด์˜ transformers๋ฅผ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋กœ ๋ถˆ๋Ÿฌ์™€์„œ ๋ชจ๋ธ์„ ์‹คํ–‰ํ•˜๋Š” ๋ฐฉ๋ฒ•์€ ์‚ฌ์šฉํ•  ์ˆ˜๋Š” ์—†์Šต๋‹ˆ๋‹ค.๐Ÿ˜‚
๊ฐ™์ด ์˜ฌ๋ ค๋†“์€ load_model.py๋ฅผ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌํ™”ํ•˜์—ฌ ๋ชจ๋ธ์„ ๋กœ๋“œํ•˜์‹œ๋ฉด ์‚ฌ์šฉ ๊ฐ€๋Šฅํ•ฉ๋‹ˆ๋‹ค.
์ถ”ํ›„ ์ง€์†์ ์ธ ์—…๋ฐ์ดํŠธ๋กœ ๋ณ„๋„์˜ ๋ถˆํŽธํ•จ ์—†์ด ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ๋„๋ก ๋ฐ”๊พธ๊ฒ ์Šต๋‹ˆ๋‹ค.
<pre><code>
# Python Code
from transformers import PreTrainedTokenizerFas
from load_model import LongformerBartForConditionalGeneration
tokenizer = PreTrainedTokenizerFast.from_pretrained("hohovisual/KoBART_with_LED")
model = LongformerBartForConditionalGeneration.from_pretrained("hohovisual/KoBART_with_LED model file path")
</pre></code>