kullm-v2-1epoch / README.md
realPCH's picture
Create README.md
ef6f8b7 verified
|
raw
history blame contribute delete
No virus
673 Bytes
metadata
license: mit
datasets:
  - nlpai-lab/kullm-v2

Developed by chPark

Training Strategy

We fine-tuned this model based on yanolja/KoSOLAR-10.7B-v0.1

Run the model

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "realPCH/ko_solra_merge"
tokenizer = AutoTokenizer.from_pretrained(model_id)

model = AutoModelForCausalLM.from_pretrained(model_id)

text = "[INST] Put instruction here. [/INST]"
inputs = tokenizer(text, return_tensors="pt")

outputs = model.generate(**inputs, max_new_tokens=20)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))