ko-morph-qwen2.5-0.5b-instruct
ko-morph-qwen2.5-0.5b-cpt๋ฅผ KoAlpaca-v1.1a๋ก ์ง์ ํ๋(SFT)ํ ๋ชจ๋ธ์ ๋๋ค. CPT ๋ชจ๋ธ ์นด๋์ ์๋ ํํ์ ํ ํฌ๋์ด์ ยท๊ฐํ ๋ถ๊ฐยท๋์กฐ๊ตฐ ์์ ๋ฑ์ ์ ์ฝ์ด ๊ทธ๋๋ก ์ ์ฉ๋ฉ๋๋ค.
โ ๏ธ ๊ณต์ Qwen2.5-0.5B-Instruct์ ๋น๊ตํ์ง ๋ง์ธ์. ๊ทธ์ชฝ์ ๋๊ท๋ชจ ๋น๊ณต๊ฐ SFT๋ฅผ ๊ฑฐ์น ๋ชจ๋ธ์ด๋ผ 21k๊ฑด์ง๋ฆฌ ํ๋(๋ณธ ๋ชจ๋ธ)๊ณผ์ ์ฐจ์ด๋ ํ ํฌ๋์ด์ ยทCPT๊ฐ ์๋๋ผ SFT ๊ท๋ชจ์์ ์ต๋๋ค. ์ด ๋ชจ๋ธ์ ๊ณต์ ํ ๋น๊ต ๋์์ ๊ฐ์ ์กฐ๊ฑด์ผ๋ก SFTํ Qwen2.5-0.5B ์๋ณธ ๋ฒ ์ด์ค์ ๋๋ค (ํ๊ฐ ๊ฒฐ๊ณผ ์๋).
SFT ์ค์
- ๋ฐ์ดํฐ: beomi/KoAlpaca-v1.1a (์ง์iN ์ง๋ฌธ + GPT ์์ฑ ๋ต๋ณ)
- ํ๋กฌํํธ ํฌ๋งท:
### ์ง๋ฌธ: {instruction} ### ๋ต๋ณ: {output}โ ๊ฐํ ์์ (ํ ํฌ๋์ด์ ๊ฐ\n์ ์ธ์ฝ๋ฉ ๋ชป ํจ) - seq 768, micro-bs 4, accum 4, 3 epoch, lr 2e-5, ํ๋กฌํํธ ๋ง์คํน (๋ต๋ณ์๋ง loss)
- Qwen ๋์กฐ๊ตฐ๊ณผ ๊ต์งํฉ ์์๋ง ์ฌ์ฉ (21,067๊ฑด/99.6%) โ ๋ ํ ํฌ๋์ด์ ์ ๊ธธ์ด ์ฐจ์ด๋ก ์ธํ ํ์ต์ ์ค์ผ ๋ฐฉ์ง
ํ๊ฐ (๋์ผ SFT ์กฐ๊ฑด์ Qwen2.5-0.5B ๋ฒ ์ด์ค์ ๋น๊ต)
| ํ๊ฐ | ๋ณธ ๋ชจ๋ธ | Qwen ์๋ณธ |
|---|---|---|
| ๋ฌธ๋ฒ ์ต์๋๋ฆฝ์ (334์) | 327/334 (97.9%) | 293/334 (87.7%) |
| ํ๊ตญ์ด ์ง์ยท์์ (8๋ฌธํญ) | ์ ํ 3 / ๋ถ๋ถ 3 / ์ค๋ต 2 | ์ค๋ต 8 |
| ์ผ๋ฐ ์ถ๋ก (10๋ฌธํญ) | ์ ๋ต 2 / ๋ถ๋ถ 1 | ์ ๋ต 3 |
McNemar p = 1.95ร10โปโธ (๋ฌธ๋ฒ ํ๊ฐ). ์ด ๊ฒฉ์ฐจ๊ฐ ํํ์ ํ ํฌ๋์ด์ ๋๋ฌธ์ธ์ง, 2.5B ํ ํฐ ํ๊ตญ์ด CPT ์์ฒด ๋๋ฌธ์ธ์ง๋ ๋ถ๋ฆฌ๋์ง ์์ต๋๋ค โ ๋์กฐ๊ตฐ(์ผ๋ฐ BPE + ๋์ผ CPT)์ด ์๊ธฐ ๋๋ฌธ์ ๋๋ค.
์ฐ์ ยทํ์๋
ผ๋ฆฌ ์ถ๋ก ์ Qwen ์๋ณธ์ด ๋ ๋ซ์ต๋๋ค (300ร4=1200 ๋ฑ). Qwen2.5๋ ์ํยท์ฝ๋๋ฅผ ๋๋
์ฌ์ ํ์ตํ๋๋ฐ ๊ทธ ์์ ํ๊ตญ์ด ์น ํ
์คํธ๋ง ์น์ CPT๋ผ catastrophic forgetting ์ ํฉ์ผ๋ก ๋ณด์
๋๋ค.
์ฌ์ฉ๋ฒ
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
tok = AutoTokenizer.from_pretrained("wwwcomcom/ko-morph-qwen2.5-0.5b-instruct")
model = AutoModelForCausalLM.from_pretrained("wwwcomcom/ko-morph-qwen2.5-0.5b-instruct", dtype=torch.float16)
prompt = "### ์ง๋ฌธ: ํ๊ตญ์ ์๋๋ ์ด๋์ผ? ### ๋ต๋ณ:"
ids = tok(prompt, return_tensors="pt").input_ids
out = model.generate(ids, max_new_tokens=128, do_sample=False)
print(tok.decode(out[0], skip_special_tokens=True))
๊ฐํ์ด ์๋ ํ๋กฌํํธ๋ ChatML ํฌ๋งท์ ์ฐ์ง ๋ง์ธ์ โ ํ์ต ๋ ๋ณธ ์ ์ด ์์ด ์ฑ๋ฅ์ด ๋จ์ด์ง๋๋ค.
๋ผ์ด์ ์ค
Qwen2.5-0.5B ๊ธฐ๋ฐ ํ์ ๋ชจ๋ธ๋ก Apache 2.0์ ๋ฐ๋ฆ ๋๋ค.
- Downloads last month
- 221