README.md

---
license: mit
datasets:
- maywell/koVast
language:
- ko
base_model:
- blueapple8259/tzem
pipeline_tag: question-answering
library_name: transformers
---
[tzem](https://huggingface.co/blueapple8259/tzem) 모델을 instruct 데이터로 파인튜닝 한 모델입니다.

## 프롬프트 템플릿

```
**사용자:** {prompt}
**인공지능:**
```

## 사용

```python
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "blueapple8259/tzem-instruct"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

prompt = "여기에 프롬프트 입력"

text = f"**사용자:** {prompt}\n**인공지능:"

inputs = tokenizer(text, return_tensors="pt")
outputs = model.generate(
    **inputs,
    max_new_tokens=100,
    pad_token_id=tokenizer.pad_token_id,
    eos_token_id=tokenizer.eos_token_id,
    do_sample=True,
    top_p=0.2,
)

output = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(output)
```

### 예시

```text
**사용자:** 인터넷 브라우저에 대해 알려줘.
**인공지능:** 인터넷 브라우저는 인터넷에서 정보를 검색하고, 다른 사용자와 소통하는 데 사용되는 소프트웨어입니다.
```

```text
**사용자:** 건강을 유지하기 위한 세 가지 팁을 알려주세요.
**인공지능:** 1. 충분한 수면을 취하세요.
2. 건강한 식단을 섭취하세요.
3. 규칙적으로 운동하세요.
```

## 데이터셋

- [maywell/koVast](https://huggingface.co/datasets/maywell/koVast)

- [KoAlpaca](https://raw.githubusercontent.com/Beomi/KoAlpaca/refs/heads/main/ko_alpaca_data.json) - 코드, 표가 포함된 데이터 제외