Views
No views yet

Update @ 2024.06.05: First release of Llama3-Ocelot-8B-instruct-v01
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("cpm-ai/Ocelot-Ko-self-instruction-10.8B-v1.0")
4model = AutoModelForCausalLM.from_pretrained("cpm-ai/Ocelot-Ko-self-instruction-10.8B-v1.0", device_map="auto")
5
6pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=4096, streamer=streamer)
7
8text = '대한민국의 수도는 어디인가요?'
9
10messages = [
11 {
12 "role": "user",
13 "content": "{}".format(text)
14 }
15]
16
17prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
18
19outputs = pipe(
20 prompt,
21 temperature=0.2,
22 add_special_tokens=True
23)
24print(outputs[0]["generated_text"][len(prompt):])
251대한민국의 수도는 서울특별시입니다.
2서울특별시에는 청와대, 국회의사당, 대법원 등 대한민국의 주요 정부기관이 위치해 있습니다.
3또한 서울시는 대한민국의 경제, 문화, 교육, 교통의 중심지로써 대한민국의 수도이자 대표 도시입니다.제가 도움이 되었길 바랍니다. 더 궁금한 점이 있으시면 언제든지 물어보세요!1@misc {cpm-ai/Ocelot-Ko-self-instruction-10.8B-v1.0,
2 author = { {frcp, nebchi, pepperonipizza97} },
3 title = { solar-kor-resume},
4 year = 2024,
5 url = { https://huggingface.co/cpm-ai/Ocelot-Ko-self-instruction-10.8B-v1.0 },
6 publisher = { Hugging Face }
7}| Model | Single turn* | Multi turn* | Overall* |
|---|---|---|---|
| gemini-1.5-pro-preview-0215 | 7.90 | 6.26 | 7.08 |
| xionic-1-72b-20240404 | 7.23 | 6.28 | 6.76 |
| Ocelot-Instruct | 6.79 | 6.71 | 6.75 |
| allganize/Llama-3-Alpha-Ko-8B-Instruct | 7.14 | 6.09 | 6.61 |