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))

realPCH/kosolra-wiki-QA-1epoch

Developed by chPark

Training Strategy

Run the model

realPCH/
kosolra-wiki-QA-1epoch