File size: 2,352 Bytes
8d65a0b
 
 
 
 
 
b847703
 
8d65a0b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d6b8e5e
8d65a0b
 
 
b847703
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
---
license: apache-2.0
language:
- ja
- en
base_model:
- TeamDelta/ABEJA-Qwen2.5-32B-base-jp-v0.1
- deepseek-ai/DeepSeek-R1-Distill-Qwen-32B
---

## 概要
このモデルは[Qwen/Qwen2.5-32B](https://huggingface.co/Qwen/Qwen2.5-32B)をファインチューニングしたAbeja社のベースモデルにDeepSeek社のR1蒸留モデルである[deepseek-ai/DeepSeek-R1-Distill-Qwen-32B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B)をChatVectorを用いて加えたものに、独自の日本語強化ファインチューニングをしたモデルとなります。

## 注意
Abeja社はbaseモデルを公開していないため、マイナスチャットベクターを用いて作りました。
このモデルは **長考モデル**ではありません。
## How to use
```python
from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "DataPilot/Arrival-32B-Instruct-v0.3"
tokenizer_name = ""

if tokenizer_name == "":
    tokenizer_name = model_name

model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype="auto",
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)

prompt = "9.9と9.11はどちらのほうが大きいですか?"
messages = [
    {"role": "system", "content": "あなたは優秀な日本語アシスタントです。問題解決をするために考えた上で回答を行ってください。"},
    {"role": "user", "content": prompt}
]
text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)

generated_ids = model.generate(
    **model_inputs,
    max_new_tokens=1024
)
generated_ids = [
    output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
]

response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]

print(response)
```

## ベンチマーク
このモデルはELYZA-task100で4.61をマークしました。(評価にはGroqのllama3-70B-8192を使用しました。)


## 謝辞
モデルの作成者であるDeepSeekチーム, Qwenチーム, Abejaチーム、評価モデルの作成者であるmeta社とAPIを公開しているGroq社、計算資源を貸していただいたVOLTMIND社に感謝を申し上げます。