Views
No views yet


| Model | COPA | HellaSwag | BoolQ | SentiNeg |
|---|---|---|---|---|
| EleutherAI/polyglot-ko-12.8b | 0.7937 | 0.5954 | 0.4818 | 0.9117 |
| Synatra-7B-v0.3-base | 0.6344 | 0.5140 | 0.5226 | NaN |
| Synatra-7B-v0.3-dpo | 0.6380 | 0.4780 | 0.8058 | 0.8942 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3device = "cuda" # the device to load the model onto
4
5model = AutoModelForCausalLM.from_pretrained("maywell/Synatra-7B-v0.3-dpo")
6tokenizer = AutoTokenizer.from_pretrained("maywell/Synatra-7B-v0.3-dpo")
7
8messages = [
9 {"role": "user", "content": "바나나는 원래 하얀색이야?"},
10]
11
12encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")
13
14model_inputs = encodeds.to(device)
15model.to(device)
16
17generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
18decoded = tokenizer.batch_decode(generated_ids)
19print(decoded[0])