kosolra-kullm / README.md
realPCH's picture
Update README.md
f40edd4 verified
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/kosolra-kullm"
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))