Yejin Korean 8B v3 SFT v4
Korean instruction-following language model.
Chat Template
<s><|user|>\n{instruction}<|end|>\n<|assistant|>\n{response}<|end|>\n</s>
Usage
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("cooler8/yejin-korean-8b-v3-sft-v4")
tokenizer = AutoTokenizer.from_pretrained("cooler8/yejin-korean-8b-v3-sft-v4")
messages = [{"role": "user", "content": "한국의 수도는 어디인가요?"}]
inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt", return_dict=True)
inputs.pop("token_type_ids", None)
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
License
Apache 2.0