model-merged / README.md
Whitewinter's picture
Upload merged fine-tuned model
cf106ac verified
|
Raw
History Blame Contribute Delete
1.56 kB
---
license: apache-2.0
base_model: Qwen/Qwen3-0.6B
tags:
- fine-tuned
- merged
- lora
- qwen
datasets:
- tatsu-lab/alpaca
language:
- ko
- en
---
# Fine-tuned Merged Model
์ด ๋ชจ๋ธ์€ Qwen/Qwen3-0.6B์„ ๊ธฐ๋ฐ˜์œผ๋กœ LoRA(Low-Rank Adaptation) ๊ธฐ๋ฒ•์„ ์‚ฌ์šฉํ•ด ํŒŒ์ธํŠœ๋‹ํ•œ ํ›„, ๊ธฐ๋ณธ ๋ชจ๋ธ๊ณผ ๋ณ‘ํ•ฉ๋œ ์™„์ „ํ•œ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.
## ๋ชจ๋ธ ์ •๋ณด
- **๋ฒ ์ด์Šค ๋ชจ๋ธ**: Qwen/Qwen3-0.6B
- **ํŒŒ์ธํŠœ๋‹ ๋ฐฉ๋ฒ•**: LoRA (Low-Rank Adaptation)
- **๋ฐ์ดํ„ฐ์…‹**: tatsu-lab/alpaca
- **๋ชจ๋ธ ํƒ€์ž…**: ์™„์ „ ๋ณ‘ํ•ฉ๋œ ๋ชจ๋ธ (Full Merged Model)
## ์‚ฌ์šฉ ๋ฐฉ๋ฒ•
```python
from transformers import AutoTokenizer, AutoModelForCausalLM
# ๋ณ‘ํ•ฉ๋œ ๋ชจ๋ธ๊ณผ ํ† ํฌ๋‚˜์ด์ € ๋กœ๋“œ (๋ณ„๋„์˜ ์–ด๋Œ‘ํ„ฐ ๋กœ๋“œ ๋ถˆํ•„์š”)
tokenizer = AutoTokenizer.from_pretrained("Whitewinter/model-merged", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
"Whitewinter/model-merged",
trust_remote_code=True,
torch_dtype=torch.bfloat16,
device_map="auto"
)
# ์ถ”๋ก 
prompt = "### Instruction:\nExplain what machine learning is.\n\n### Response:\n"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.7)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)
```
## ํŠน์ง•
- LoRA ์–ด๋Œ‘ํ„ฐ๊ฐ€ ๊ธฐ๋ณธ ๋ชจ๋ธ๊ณผ ์™„์ „ํžˆ ๋ณ‘ํ•ฉ๋˜์–ด ์žˆ์–ด ๋ณ„๋„์˜ ์–ด๋Œ‘ํ„ฐ ๋กœ๋“œ๊ฐ€ ๋ถˆํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.
- ํ‘œ์ค€ transformers ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋งŒ์œผ๋กœ ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
- ์ถ”๋ก  ์‹œ ์†๋„๊ฐ€ ๋น ๋ฅด๊ณ  ๋ฉ”๋ชจ๋ฆฌ ํšจ์œจ์ ์ž…๋‹ˆ๋‹ค.