Yi-Ko-6B-Instruct-v1.1
Model Details
Base Model
beomi/Yi-Ko-6B
Training Dataset
- kyujinpy/KOR-OpenOrca-Platypus-v3 🙇
- beomi/KoAlpaca-v1.1a 🙇
- maywell/ko_wikidata_QA 🙇
- AIHub 데이터 활용
Instruction Format
{instruction}
{response}
Loading the Model
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("wkshin89/Yi-Ko-6B-Instruct-v1.1")
model = AutoModelForCausalLM.from_pretrained(
"wkshin89/Yi-Ko-6B-Instruct-v1.1",
device_map="auto",
torch_dtype=torch.bfloat16,
)