kurogane commited on
Commit
118293d
·
verified ·
1 Parent(s): b4f9554

Upload 12 files

Browse files
README.md CHANGED
@@ -1,12 +1,28 @@
1
  ---
2
- license: mit
3
- language:
4
- - ja
5
- base_model:
6
- - sbintuitions/sarashina2.2-3b-instruct-v0.1
 
7
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
 
9
- 以下の設定でdepth up-scalingを行ったモデルです。
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
+ }