Upload 12 files
Browse files- README.md +22 -93
- config.json +30 -0
- config_sara_depth1.yml +25 -0
- mergekit_config.yml +25 -0
- model-00001-of-00003.safetensors +3 -0
- model-00002-of-00003.safetensors +3 -0
- model-00003-of-00003.safetensors +3 -0
- model.safetensors.index.json +1 -0
- model_run_test1.ipynb +180 -0
- special_tokens_map.json +51 -0
- tokenizer.model +3 -0
- tokenizer_config.json +171 -0
README.md
CHANGED
@@ -1,12 +1,28 @@
|
|
1 |
---
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
-
|
|
|
7 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
|
9 |
-
|
10 |
|
11 |
```yaml
|
12 |
slices:
|
@@ -34,92 +50,5 @@ slices:
|
|
34 |
|
35 |
merge_method: passthrough
|
36 |
dtype: bfloat16
|
37 |
-
```
|
38 |
-
|
39 |
-
何故かぱっと見は正常な出力が得られます。
|
40 |
-
```python
|
41 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer
|
42 |
-
import torch
|
43 |
-
|
44 |
-
dir_model = r"/media/kurogane/HD-NRLD-A/novel_train/satashina_1/sara_depth1"
|
45 |
-
|
46 |
-
tokenizer = AutoTokenizer.from_pretrained(dir_model)
|
47 |
-
|
48 |
-
model = AutoModelForCausalLM.from_pretrained(
|
49 |
-
dir_model,
|
50 |
-
torch_dtype=torch.bfloat16,
|
51 |
-
device_map="auto",
|
52 |
-
)
|
53 |
-
|
54 |
-
|
55 |
-
# チャット形式での使用例
|
56 |
-
messages = [
|
57 |
-
{
|
58 |
-
"role": "system",
|
59 |
-
"content": "あなたは親切で有能なAIアシスタントです。"
|
60 |
-
},
|
61 |
-
{
|
62 |
-
"role": "user",
|
63 |
-
"content": "次の数学の問題を解いてください:2x + 3 = 7"
|
64 |
-
},
|
65 |
-
]
|
66 |
-
|
67 |
-
|
68 |
-
|
69 |
-
# チャットテンプレートを使用してメッセージを整形
|
70 |
-
prompt = tokenizer.apply_chat_template(
|
71 |
-
messages,
|
72 |
-
tokenize=False,
|
73 |
-
)
|
74 |
-
print("\\nGenerated prompt:\\n", prompt)
|
75 |
-
|
76 |
-
# トークン化と推論
|
77 |
-
inputs = tokenizer(
|
78 |
-
prompt,
|
79 |
-
return_tensors="pt",
|
80 |
-
max_length=2048,
|
81 |
-
truncation=True,
|
82 |
-
).to(model.device)
|
83 |
-
|
84 |
-
with torch.no_grad():
|
85 |
-
output = model.generate(
|
86 |
-
input_ids=inputs['input_ids'],
|
87 |
-
attention_mask=inputs['attention_mask'],
|
88 |
-
max_new_tokens=300,
|
89 |
-
do_sample=True,
|
90 |
-
top_p=0.95,
|
91 |
-
temperature=0.7,
|
92 |
-
repetition_penalty=1.05,
|
93 |
-
)
|
94 |
-
|
95 |
-
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
96 |
-
print(response)
|
97 |
-
```
|
98 |
|
99 |
```
|
100 |
-
Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.
|
101 |
-
あなたは親切で有能なAIアシスタントです。次の数学の問題を解いてください:2x + 3 = 7
|
102 |
-
(ただし、xは整数であると仮定します。)"
|
103 |
-
もちろんです、お手伝いします!
|
104 |
-
|
105 |
-
与えられた方程式は 2x + 3 = 7 で、xは整数(integer)であると仮定しています。まず、xを片側に孤立させるために、両辺から3を引いてみましょう。
|
106 |
-
|
107 |
-
両辺から3を引くと:
|
108 |
-
2x + 3 - 3 = 7 - 3
|
109 |
-
|
110 |
-
これにより、
|
111 |
-
2x = 4
|
112 |
-
|
113 |
-
次に、両辺を2で割ってxを求めます:
|
114 |
-
2x / 2 = 4 / 2
|
115 |
-
|
116 |
-
これにより、
|
117 |
-
x = 2
|
118 |
-
|
119 |
-
確認のために、元の方程式に x = 2 を代入してみましょう:
|
120 |
-
2 * 2 + 3 = 7
|
121 |
-
4 + 3 = 7
|
122 |
-
7 = 7
|
123 |
-
|
124 |
-
確かに、この解は正しいことが確認できました。したがって、xの値は2です。
|
125 |
-
```
|
|
|
1 |
---
|
2 |
+
base_model: []
|
3 |
+
library_name: transformers
|
4 |
+
tags:
|
5 |
+
- mergekit
|
6 |
+
- merge
|
7 |
+
|
8 |
---
|
9 |
+
# sara_depth1
|
10 |
+
|
11 |
+
This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
|
12 |
+
|
13 |
+
## Merge Details
|
14 |
+
### Merge Method
|
15 |
+
|
16 |
+
This model was merged using the Passthrough merge method.
|
17 |
+
|
18 |
+
### Models Merged
|
19 |
+
|
20 |
+
The following models were included in the merge:
|
21 |
+
* sarashina2.2-3b-instruct-v0.1
|
22 |
+
|
23 |
+
### Configuration
|
24 |
|
25 |
+
The following YAML configuration was used to produce this model:
|
26 |
|
27 |
```yaml
|
28 |
slices:
|
|
|
50 |
|
51 |
merge_method: passthrough
|
52 |
dtype: bfloat16
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
|
54 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config.json
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "sarashina2.2-3b-instruct-v0.1",
|
3 |
+
"architectures": [
|
4 |
+
"LlamaForCausalLM"
|
5 |
+
],
|
6 |
+
"attention_bias": false,
|
7 |
+
"attention_dropout": 0.0,
|
8 |
+
"bos_token_id": 1,
|
9 |
+
"eos_token_id": 2,
|
10 |
+
"head_dim": 160,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 2560,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 8960,
|
15 |
+
"max_position_embeddings": 8192,
|
16 |
+
"mlp_bias": false,
|
17 |
+
"model_type": "llama",
|
18 |
+
"num_attention_heads": 16,
|
19 |
+
"num_hidden_layers": 56,
|
20 |
+
"num_key_value_heads": 8,
|
21 |
+
"pretraining_tp": 1,
|
22 |
+
"rms_norm_eps": 1e-05,
|
23 |
+
"rope_scaling": null,
|
24 |
+
"rope_theta": 500000,
|
25 |
+
"tie_word_embeddings": false,
|
26 |
+
"torch_dtype": "bfloat16",
|
27 |
+
"transformers_version": "4.48.3",
|
28 |
+
"use_cache": false,
|
29 |
+
"vocab_size": 102400
|
30 |
+
}
|
config_sara_depth1.yml
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
slices:
|
2 |
+
- sources:
|
3 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
4 |
+
layer_range: [0, 8]
|
5 |
+
- sources:
|
6 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
7 |
+
layer_range: [4, 12]
|
8 |
+
- sources:
|
9 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
10 |
+
layer_range: [8, 16]
|
11 |
+
- sources:
|
12 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
13 |
+
layer_range: [12, 20]
|
14 |
+
- sources:
|
15 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
16 |
+
layer_range: [16, 24]
|
17 |
+
- sources:
|
18 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
19 |
+
layer_range: [20, 28]
|
20 |
+
- sources:
|
21 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
22 |
+
layer_range: [24, 32]
|
23 |
+
|
24 |
+
merge_method: passthrough
|
25 |
+
dtype: bfloat16
|
mergekit_config.yml
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
slices:
|
2 |
+
- sources:
|
3 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
4 |
+
layer_range: [0, 8]
|
5 |
+
- sources:
|
6 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
7 |
+
layer_range: [4, 12]
|
8 |
+
- sources:
|
9 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
10 |
+
layer_range: [8, 16]
|
11 |
+
- sources:
|
12 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
13 |
+
layer_range: [12, 20]
|
14 |
+
- sources:
|
15 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
16 |
+
layer_range: [16, 24]
|
17 |
+
- sources:
|
18 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
19 |
+
layer_range: [20, 28]
|
20 |
+
- sources:
|
21 |
+
- model: 'sarashina2.2-3b-instruct-v0.1'
|
22 |
+
layer_range: [24, 32]
|
23 |
+
|
24 |
+
merge_method: passthrough
|
25 |
+
dtype: bfloat16
|
model-00001-of-00003.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:34c575b6099c7999a94d961d137a20429130a81757584c9bf44fe8e26d54322c
|
3 |
+
size 4987543496
|
model-00002-of-00003.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5ad6e6f9c8d43bd744b7a94ec81f4797dabaf84ef470f7ea059bfc428cc91e41
|
3 |
+
size 4954842760
|
model-00003-of-00003.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:46f5f148cc5ab981c689688d2a248c2a48994eb033203d3fe65b9f5af5bafd61
|
3 |
+
size 1015870280
|
model.safetensors.index.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"metadata": {"mergekit_version": "0.1.1"}, "weight_map": {"lm_head.weight": "model-00001-of-00003.safetensors", "model.embed_tokens.weight": "model-00001-of-00003.safetensors", "model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.0.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.0.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.1.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.14.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.18.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.15.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.19.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.20.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.24.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.20.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.24.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.24.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.21.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.25.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.21.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.25.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.25.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.21.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.22.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.26.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.22.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.26.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.26.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.22.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.23.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.27.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.23.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.27.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.27.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.23.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.28.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.32.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.28.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.32.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.32.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.28.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.29.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.33.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.29.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.33.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.33.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.29.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.30.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.34.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.30.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.34.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.34.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.30.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.31.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.35.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.mlp.gate_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.mlp.up_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.31.post_attention_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.35.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.self_attn.k_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.self_attn.o_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.self_attn.q_proj.weight": "model-00001-of-00003.safetensors", "model.layers.35.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.31.self_attn.v_proj.weight": "model-00001-of-00003.safetensors", "model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors", "model.layers.2.mlp.down_proj.weight": "model-00001-of-00003.safetensors", "model.layers.2.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.2.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.2.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.2.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.2.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.2.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.2.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.36.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.40.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.36.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.40.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.40.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.36.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.37.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.41.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.37.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.41.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.41.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.37.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.38.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.42.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.38.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.42.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.42.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.38.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.39.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.43.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.39.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.43.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.43.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.39.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.44.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.48.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.44.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.48.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.48.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.44.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.45.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.49.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.45.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.49.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.49.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.45.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.46.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.50.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.46.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.50.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.50.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.46.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.47.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.51.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.47.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.51.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.51.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.47.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.52.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.52.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.52.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.53.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.53.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.53.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.3.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.54.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.54.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.54.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.55.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.55.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.55.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.4.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.8.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.5.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.9.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.6.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.10.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.mlp.up_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00003.safetensors", "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00003.safetensors", "model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.7.input_layernorm.weight": "model-00002-of-00003.safetensors", "model.layers.11.mlp.down_proj.weight": "model-00002-of-00003.safetensors", "model.layers.7.mlp.down_proj.weight": "model-00003-of-00003.safetensors", "model.layers.11.mlp.gate_proj.weight": "model-00003-of-00003.safetensors", "model.layers.7.mlp.gate_proj.weight": "model-00003-of-00003.safetensors", "model.layers.11.mlp.up_proj.weight": "model-00003-of-00003.safetensors", "model.layers.7.mlp.up_proj.weight": "model-00003-of-00003.safetensors", "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.7.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.11.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", "model.layers.7.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", "model.layers.11.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", "model.layers.7.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", "model.layers.11.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", "model.layers.7.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", "model.layers.11.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", "model.layers.7.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.input_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.12.input_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.16.mlp.down_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.mlp.down_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.mlp.gate_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.mlp.gate_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.mlp.up_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.mlp.up_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.16.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", "model.layers.16.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", "model.layers.12.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.input_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.13.input_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.17.mlp.down_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.mlp.down_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.mlp.gate_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.mlp.gate_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.mlp.up_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.mlp.up_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.13.post_attention_layernorm.weight": "model-00003-of-00003.safetensors", "model.layers.17.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.self_attn.k_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.self_attn.q_proj.weight": "model-00003-of-00003.safetensors", "model.layers.17.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", "model.layers.13.self_attn.v_proj.weight": "model-00003-of-00003.safetensors", "model.norm.weight": "model-00003-of-00003.safetensors"}}
|
model_run_test1.ipynb
ADDED
@@ -0,0 +1,180 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"cells": [
|
3 |
+
{
|
4 |
+
"cell_type": "code",
|
5 |
+
"execution_count": 12,
|
6 |
+
"metadata": {},
|
7 |
+
"outputs": [
|
8 |
+
{
|
9 |
+
"data": {
|
10 |
+
"application/vnd.jupyter.widget-view+json": {
|
11 |
+
"model_id": "1fd6faeab6d64b36bd96bb1bf64c452a",
|
12 |
+
"version_major": 2,
|
13 |
+
"version_minor": 0
|
14 |
+
},
|
15 |
+
"text/plain": [
|
16 |
+
"Loading checkpoint shards: 0%| | 0/3 [00:00<?, ?it/s]"
|
17 |
+
]
|
18 |
+
},
|
19 |
+
"metadata": {},
|
20 |
+
"output_type": "display_data"
|
21 |
+
},
|
22 |
+
{
|
23 |
+
"name": "stderr",
|
24 |
+
"output_type": "stream",
|
25 |
+
"text": [
|
26 |
+
"Some parameters are on the meta device because they were offloaded to the cpu.\n"
|
27 |
+
]
|
28 |
+
}
|
29 |
+
],
|
30 |
+
"source": [
|
31 |
+
"from transformers import AutoModelForCausalLM, AutoTokenizer\n",
|
32 |
+
"import torch\n",
|
33 |
+
"\n",
|
34 |
+
"dir_model = r\"/media/kurogane/HD-NRLD-A/novel_train/satashina_1/sara_depth1\"\n",
|
35 |
+
"\n",
|
36 |
+
"tokenizer = AutoTokenizer.from_pretrained(dir_model)\n",
|
37 |
+
"\n",
|
38 |
+
"model = AutoModelForCausalLM.from_pretrained(\n",
|
39 |
+
" dir_model, \n",
|
40 |
+
" torch_dtype=torch.bfloat16, \n",
|
41 |
+
" device_map=\"auto\",\n",
|
42 |
+
")\n"
|
43 |
+
]
|
44 |
+
},
|
45 |
+
{
|
46 |
+
"cell_type": "code",
|
47 |
+
"execution_count": 13,
|
48 |
+
"metadata": {},
|
49 |
+
"outputs": [
|
50 |
+
{
|
51 |
+
"name": "stdout",
|
52 |
+
"output_type": "stream",
|
53 |
+
"text": [
|
54 |
+
"\\nGenerated prompt:\\n <|system|>あなたは親切で有能なAIアシスタントです。</s><|user|>次の数学の問題を解いてください:2x + 3 = 7</s>\n"
|
55 |
+
]
|
56 |
+
}
|
57 |
+
],
|
58 |
+
"source": [
|
59 |
+
"\n",
|
60 |
+
"\n",
|
61 |
+
"# チャット形式での使用例\n",
|
62 |
+
"messages = [\n",
|
63 |
+
" {\n",
|
64 |
+
" \"role\": \"system\",\n",
|
65 |
+
" \"content\": \"あなたは親切で有能なAIアシスタントです。\"\n",
|
66 |
+
" },\n",
|
67 |
+
" {\n",
|
68 |
+
" \"role\": \"user\",\n",
|
69 |
+
" \"content\": \"次の数学の問題を解いてください:2x + 3 = 7\"\n",
|
70 |
+
" },\n",
|
71 |
+
"]\n",
|
72 |
+
"\n",
|
73 |
+
"\n",
|
74 |
+
"\n",
|
75 |
+
"# チャットテンプレートを使用してメッセージを整形\n",
|
76 |
+
"prompt = tokenizer.apply_chat_template(\n",
|
77 |
+
" messages, \n",
|
78 |
+
" tokenize=False,\n",
|
79 |
+
" )\n",
|
80 |
+
"print(\"\\\\nGenerated prompt:\\\\n\", prompt)\n",
|
81 |
+
"\n",
|
82 |
+
"# トークン化と推論\n",
|
83 |
+
"inputs = tokenizer(\n",
|
84 |
+
" prompt, \n",
|
85 |
+
" return_tensors=\"pt\", \n",
|
86 |
+
" max_length=2048, \n",
|
87 |
+
" truncation=True,\n",
|
88 |
+
" ).to(model.device)\n",
|
89 |
+
"\n"
|
90 |
+
]
|
91 |
+
},
|
92 |
+
{
|
93 |
+
"cell_type": "code",
|
94 |
+
"execution_count": 15,
|
95 |
+
"metadata": {},
|
96 |
+
"outputs": [
|
97 |
+
{
|
98 |
+
"name": "stderr",
|
99 |
+
"output_type": "stream",
|
100 |
+
"text": [
|
101 |
+
"Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
|
102 |
+
]
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"name": "stdout",
|
106 |
+
"output_type": "stream",
|
107 |
+
"text": [
|
108 |
+
"あなたは親切で有能なAIアシスタントです。次の数学の問題を解いてください:2x + 3 = 7\n",
|
109 |
+
" (ただし、xは整数であると仮定します。)\"\n",
|
110 |
+
"もちろんです、お手伝いします!\n",
|
111 |
+
"\n",
|
112 |
+
"与えられた方程式は 2x + 3 = 7 で、xは整数(integer)であると仮定しています。まず、xを片側に孤立させるために、両辺から3を引いてみましょう。\n",
|
113 |
+
"\n",
|
114 |
+
"両辺から3を引くと:\n",
|
115 |
+
"2x + 3 - 3 = 7 - 3\n",
|
116 |
+
"\n",
|
117 |
+
"これにより、\n",
|
118 |
+
"2x = 4\n",
|
119 |
+
"\n",
|
120 |
+
"次に、両辺を2で割ってxを求めます:\n",
|
121 |
+
"2x / 2 = 4 / 2\n",
|
122 |
+
"\n",
|
123 |
+
"これにより、\n",
|
124 |
+
"x = 2\n",
|
125 |
+
"\n",
|
126 |
+
"確認のために、元の方程式に x = 2 を代入してみましょう:\n",
|
127 |
+
"2 * 2 + 3 = 7\n",
|
128 |
+
"4 + 3 = 7\n",
|
129 |
+
"7 = 7\n",
|
130 |
+
"\n",
|
131 |
+
"確かに、この解は正しいことが確認できました。したがって、xの値は2です。\n"
|
132 |
+
]
|
133 |
+
}
|
134 |
+
],
|
135 |
+
"source": [
|
136 |
+
"with torch.no_grad():\n",
|
137 |
+
" output = model.generate(\n",
|
138 |
+
" input_ids=inputs['input_ids'],\n",
|
139 |
+
" attention_mask=inputs['attention_mask'],\n",
|
140 |
+
" max_new_tokens=300,\n",
|
141 |
+
" do_sample=True,\n",
|
142 |
+
" top_p=0.95,\n",
|
143 |
+
" temperature=0.7,\n",
|
144 |
+
" repetition_penalty=1.05,\n",
|
145 |
+
" )\n",
|
146 |
+
"\n",
|
147 |
+
"response = tokenizer.decode(output[0], skip_special_tokens=True)\n",
|
148 |
+
"print(response)"
|
149 |
+
]
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"cell_type": "code",
|
153 |
+
"execution_count": null,
|
154 |
+
"metadata": {},
|
155 |
+
"outputs": [],
|
156 |
+
"source": []
|
157 |
+
}
|
158 |
+
],
|
159 |
+
"metadata": {
|
160 |
+
"kernelspec": {
|
161 |
+
"display_name": "vllmtest",
|
162 |
+
"language": "python",
|
163 |
+
"name": "python3"
|
164 |
+
},
|
165 |
+
"language_info": {
|
166 |
+
"codemirror_mode": {
|
167 |
+
"name": "ipython",
|
168 |
+
"version": 3
|
169 |
+
},
|
170 |
+
"file_extension": ".py",
|
171 |
+
"mimetype": "text/x-python",
|
172 |
+
"name": "python",
|
173 |
+
"nbconvert_exporter": "python",
|
174 |
+
"pygments_lexer": "ipython3",
|
175 |
+
"version": "3.12.4"
|
176 |
+
}
|
177 |
+
},
|
178 |
+
"nbformat": 4,
|
179 |
+
"nbformat_minor": 2
|
180 |
+
}
|
special_tokens_map.json
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token": {
|
3 |
+
"content": "<s>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"cls_token": {
|
10 |
+
"content": "<cls>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"eos_token": {
|
17 |
+
"content": "</s>",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
},
|
23 |
+
"mask_token": {
|
24 |
+
"content": "<mask>",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
},
|
30 |
+
"pad_token": {
|
31 |
+
"content": "<pad>",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false
|
36 |
+
},
|
37 |
+
"sep_token": {
|
38 |
+
"content": "<sep>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false
|
43 |
+
},
|
44 |
+
"unk_token": {
|
45 |
+
"content": "<unk>",
|
46 |
+
"lstrip": false,
|
47 |
+
"normalized": false,
|
48 |
+
"rstrip": false,
|
49 |
+
"single_word": false
|
50 |
+
}
|
51 |
+
}
|
tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:008293028e1a9d9a1038d9b63d989a2319797dfeaa03f171093a57b33a3a8277
|
3 |
+
size 1831879
|
tokenizer_config.json
ADDED
@@ -0,0 +1,171 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_dummy_prefix_space": false,
|
4 |
+
"add_eos_token": false,
|
5 |
+
"add_prefix_space": false,
|
6 |
+
"added_tokens_decoder": {
|
7 |
+
"0": {
|
8 |
+
"content": "<unk>",
|
9 |
+
"lstrip": false,
|
10 |
+
"normalized": false,
|
11 |
+
"rstrip": false,
|
12 |
+
"single_word": false,
|
13 |
+
"special": true
|
14 |
+
},
|
15 |
+
"1": {
|
16 |
+
"content": "<s>",
|
17 |
+
"lstrip": false,
|
18 |
+
"normalized": false,
|
19 |
+
"rstrip": false,
|
20 |
+
"single_word": false,
|
21 |
+
"special": true
|
22 |
+
},
|
23 |
+
"2": {
|
24 |
+
"content": "</s>",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false,
|
29 |
+
"special": true
|
30 |
+
},
|
31 |
+
"3": {
|
32 |
+
"content": "<pad>",
|
33 |
+
"lstrip": false,
|
34 |
+
"normalized": false,
|
35 |
+
"rstrip": false,
|
36 |
+
"single_word": false,
|
37 |
+
"special": true
|
38 |
+
},
|
39 |
+
"4": {
|
40 |
+
"content": "<sep>",
|
41 |
+
"lstrip": false,
|
42 |
+
"normalized": false,
|
43 |
+
"rstrip": false,
|
44 |
+
"single_word": false,
|
45 |
+
"special": true
|
46 |
+
},
|
47 |
+
"5": {
|
48 |
+
"content": "<mask>",
|
49 |
+
"lstrip": false,
|
50 |
+
"normalized": false,
|
51 |
+
"rstrip": false,
|
52 |
+
"single_word": false,
|
53 |
+
"special": true
|
54 |
+
},
|
55 |
+
"6": {
|
56 |
+
"content": "<cls>",
|
57 |
+
"lstrip": false,
|
58 |
+
"normalized": false,
|
59 |
+
"rstrip": false,
|
60 |
+
"single_word": false,
|
61 |
+
"special": true
|
62 |
+
},
|
63 |
+
"7": {
|
64 |
+
"content": "<|system|>",
|
65 |
+
"lstrip": false,
|
66 |
+
"normalized": false,
|
67 |
+
"rstrip": false,
|
68 |
+
"single_word": false,
|
69 |
+
"special": false
|
70 |
+
},
|
71 |
+
"8": {
|
72 |
+
"content": "<|assistant|>",
|
73 |
+
"lstrip": false,
|
74 |
+
"normalized": false,
|
75 |
+
"rstrip": false,
|
76 |
+
"single_word": false,
|
77 |
+
"special": false
|
78 |
+
},
|
79 |
+
"9": {
|
80 |
+
"content": "<|user|>",
|
81 |
+
"lstrip": false,
|
82 |
+
"normalized": false,
|
83 |
+
"rstrip": false,
|
84 |
+
"single_word": false,
|
85 |
+
"special": false
|
86 |
+
},
|
87 |
+
"10": {
|
88 |
+
"content": "<|available_tools|>",
|
89 |
+
"lstrip": false,
|
90 |
+
"normalized": false,
|
91 |
+
"rstrip": false,
|
92 |
+
"single_word": false,
|
93 |
+
"special": false
|
94 |
+
},
|
95 |
+
"11": {
|
96 |
+
"content": "<|tool_calls|>",
|
97 |
+
"lstrip": false,
|
98 |
+
"normalized": false,
|
99 |
+
"rstrip": false,
|
100 |
+
"single_word": false,
|
101 |
+
"special": false
|
102 |
+
},
|
103 |
+
"12": {
|
104 |
+
"content": "<|tool_results|>",
|
105 |
+
"lstrip": false,
|
106 |
+
"normalized": false,
|
107 |
+
"rstrip": false,
|
108 |
+
"single_word": false,
|
109 |
+
"special": false
|
110 |
+
},
|
111 |
+
"13": {
|
112 |
+
"content": "<|code|>",
|
113 |
+
"lstrip": false,
|
114 |
+
"normalized": false,
|
115 |
+
"rstrip": false,
|
116 |
+
"single_word": false,
|
117 |
+
"special": false
|
118 |
+
},
|
119 |
+
"14": {
|
120 |
+
"content": "<|file|>",
|
121 |
+
"lstrip": false,
|
122 |
+
"normalized": false,
|
123 |
+
"rstrip": false,
|
124 |
+
"single_word": false,
|
125 |
+
"special": false
|
126 |
+
},
|
127 |
+
"102397": {
|
128 |
+
"content": "<|prefix|>",
|
129 |
+
"lstrip": false,
|
130 |
+
"normalized": false,
|
131 |
+
"rstrip": false,
|
132 |
+
"single_word": false,
|
133 |
+
"special": false
|
134 |
+
},
|
135 |
+
"102398": {
|
136 |
+
"content": "<|suffix|>",
|
137 |
+
"lstrip": false,
|
138 |
+
"normalized": false,
|
139 |
+
"rstrip": false,
|
140 |
+
"single_word": false,
|
141 |
+
"special": false
|
142 |
+
},
|
143 |
+
"102399": {
|
144 |
+
"content": "<|middle|>",
|
145 |
+
"lstrip": false,
|
146 |
+
"normalized": false,
|
147 |
+
"rstrip": false,
|
148 |
+
"single_word": false,
|
149 |
+
"special": false
|
150 |
+
}
|
151 |
+
},
|
152 |
+
"bos_token": "<s>",
|
153 |
+
"chat_template": "\n{%- set user_messages = messages | selectattr('role', 'equalto', 'user') | list %}\n{%- macro output_available_tools(tools, message) %}\n{%- if tools and (message == user_messages[-1]) %}\n {{- '<|available_tools|>[' }}\n {%- for tool in tools %}\n {%- set tool = tool.function %}\n {{- \"{\" }}\n {%- for key, val in tool.items() if key != \"return\" %}\n {%- if val is string %}\n {{- \"'\" + key + \"': '\" + val + \"'\" }}\n {%- else %}\n {{- \"'\" + key + \"': \" + val|string }}\n {%- endif %}\n {%- if not loop.last %}\n {{- \", \" }}\n {%- endif %}\n {%- endfor %}\n {{- \"}\" }}\n {%- if not loop.last %}\n {{- \", \" }}\n {%- else %}\n {{- \"]\" }}\n {%- endif %}\n {%- endfor %}\n {{- eos_token -}}\n{%- endif %}\n{%- endmacro %}\n\n{%- macro output_tool_results(tool_results) %}\n{{- '<|tool_results|>[' }}\n{%- for tool_result in tool_results %}\n {{- \"{'content': \" + tool_result.content|string + \", 'call_id': '\" + tool_result.call_id + \"'}\" }}\n{%- endfor %}\n{{- ']' }}\n{{- eos_token -}}\n{%- endmacro %}\n\n{%- macro output_tool_calls(tool_calls) %}\n{{- '<|tool_calls|>[' }}\n{%- for tool_call in tool_calls %}\n {{- \"{'id': '\" + tool_call.id + \"', 'name': '\" + tool_call.name + \"', 'arguments': \" + tool_call.arguments|string + '}' }}\n{%- endfor %}\n{{- ']' }}\n{%- endmacro %}\n\n{%- for message in messages %}\n {%- if message['role'] == 'user' %}\n {%- if tools is defined %}\n {{- output_available_tools(tools, message) }}\n {%- endif %}\n {{- '<|user|>' + message['content'] + eos_token -}}\n {%- elif message['role'] == 'system' %}\n {{- '<|system|>' + message['content'] + eos_token -}}\n {%- elif message['role'] == 'assistant' %}\n {% set assistant_content = \"\" %}\n {%- if message.content is defined %}\n {% set assistant_content = message.content %}\n {%- endif %}\n {%- if message.tool_calls is defined and message.tool_calls -%}\n {{- '<|assistant|>' + assistant_content + output_tool_calls(message['tool_calls']) + eos_token -}}\n {%- else %}\n {{- '<|assistant|>' + assistant_content + eos_token }}\n {%- endif %}\n {%- elif message['role'] == 'tool_results' %}\n {{- output_tool_results(message.tool_results) }}\n {%- endif %}\n{%- if loop.last and add_generation_prompt -%}\n {{- '<|assistant|>' -}}\n{%- endif -%}\n{%- endfor %}\n",
|
154 |
+
"clean_up_tokenization_spaces": false,
|
155 |
+
"cls_token": "<cls>",
|
156 |
+
"do_lower_case": false,
|
157 |
+
"eos_token": "</s>",
|
158 |
+
"extra_ids": 0,
|
159 |
+
"keep_accents": true,
|
160 |
+
"legacy": false,
|
161 |
+
"mask_token": "<mask>",
|
162 |
+
"model_max_length": 1000000000000000019884624838656,
|
163 |
+
"pad_token": "<pad>",
|
164 |
+
"padding_side": "right",
|
165 |
+
"sep_token": "<sep>",
|
166 |
+
"sp_model_kwargs": {},
|
167 |
+
"spaces_between_special_tokens": false,
|
168 |
+
"tokenizer_class": "LlamaTokenizer",
|
169 |
+
"unk_token": "<unk>",
|
170 |
+
"use_default_system_prompt": false
|
171 |
+
}
|