phpaiola commited on
Commit
56c0c97
1 Parent(s): 3e9f359

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/",
3
+ "architectures": [
4
+ "GemmaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 2,
9
+ "eos_token_id": 1,
10
+ "head_dim": 256,
11
+ "hidden_act": "gelu",
12
+ "hidden_activation": null,
13
+ "hidden_size": 2048,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 16384,
16
+ "max_position_embeddings": 8192,
17
+ "model_type": "gemma",
18
+ "num_attention_heads": 8,
19
+ "num_hidden_layers": 18,
20
+ "num_key_value_heads": 1,
21
+ "pad_token_id": 0,
22
+ "rms_norm_eps": 1e-06,
23
+ "rope_scaling": null,
24
+ "rope_theta": 10000.0,
25
+ "torch_dtype": "float16",
26
+ "transformers_version": "4.39.1",
27
+ "use_cache": false,
28
+ "vocab_size": 256000
29
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": 1,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "4.39.1"
7
+ }
pytorch_model-00001-of-00003.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0c6c28e533ee19ccbaf5e1dcdd7f9edbd791ceedd19587dd4df64c56cb59337
3
+ size 1948301196
pytorch_model-00002-of-00003.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d596f08f2f579b852bc7760a63b9f951ce1c756b90b821e75b74aa98227e9d8d
3
+ size 1981910044
pytorch_model-00003-of-00003.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:caaee167b8ce5bf51bd2473074c2093d4be9197f4bc1665112d58afd109ace46
3
+ size 1082189982
pytorch_model.bin.index.json ADDED
@@ -0,0 +1,172 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 5012344832
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "pytorch_model-00001-of-00003.bin",
7
+ "model.embed_tokens.weight": "pytorch_model-00001-of-00003.bin",
8
+ "model.layers.0.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
9
+ "model.layers.0.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
10
+ "model.layers.0.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
11
+ "model.layers.0.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
12
+ "model.layers.0.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
13
+ "model.layers.0.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
14
+ "model.layers.0.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
15
+ "model.layers.0.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
16
+ "model.layers.0.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
17
+ "model.layers.1.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
18
+ "model.layers.1.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
19
+ "model.layers.1.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
20
+ "model.layers.1.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
21
+ "model.layers.1.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
22
+ "model.layers.1.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
23
+ "model.layers.1.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
24
+ "model.layers.1.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
25
+ "model.layers.1.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
26
+ "model.layers.10.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
27
+ "model.layers.10.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
28
+ "model.layers.10.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
29
+ "model.layers.10.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
30
+ "model.layers.10.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
31
+ "model.layers.10.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
32
+ "model.layers.10.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
33
+ "model.layers.10.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
34
+ "model.layers.10.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
35
+ "model.layers.11.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
36
+ "model.layers.11.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
37
+ "model.layers.11.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
38
+ "model.layers.11.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
39
+ "model.layers.11.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
40
+ "model.layers.11.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
41
+ "model.layers.11.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
42
+ "model.layers.11.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
43
+ "model.layers.11.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
44
+ "model.layers.12.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
45
+ "model.layers.12.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
46
+ "model.layers.12.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
47
+ "model.layers.12.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
48
+ "model.layers.12.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
49
+ "model.layers.12.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
50
+ "model.layers.12.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
51
+ "model.layers.12.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
52
+ "model.layers.12.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
53
+ "model.layers.13.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
54
+ "model.layers.13.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
55
+ "model.layers.13.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
56
+ "model.layers.13.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
57
+ "model.layers.13.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
58
+ "model.layers.13.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
59
+ "model.layers.13.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
60
+ "model.layers.13.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
61
+ "model.layers.13.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
62
+ "model.layers.14.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
63
+ "model.layers.14.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
64
+ "model.layers.14.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
65
+ "model.layers.14.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
66
+ "model.layers.14.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
67
+ "model.layers.14.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
68
+ "model.layers.14.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
69
+ "model.layers.14.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
70
+ "model.layers.14.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
71
+ "model.layers.15.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
72
+ "model.layers.15.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
73
+ "model.layers.15.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
74
+ "model.layers.15.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
75
+ "model.layers.15.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
76
+ "model.layers.15.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
77
+ "model.layers.15.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
78
+ "model.layers.15.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
79
+ "model.layers.15.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
80
+ "model.layers.16.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
81
+ "model.layers.16.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
82
+ "model.layers.16.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
83
+ "model.layers.16.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
84
+ "model.layers.16.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
85
+ "model.layers.16.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
86
+ "model.layers.16.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
87
+ "model.layers.16.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
88
+ "model.layers.16.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
89
+ "model.layers.17.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
90
+ "model.layers.17.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
91
+ "model.layers.17.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
92
+ "model.layers.17.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
93
+ "model.layers.17.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
94
+ "model.layers.17.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
95
+ "model.layers.17.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
96
+ "model.layers.17.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
97
+ "model.layers.17.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
98
+ "model.layers.2.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
99
+ "model.layers.2.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
100
+ "model.layers.2.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
101
+ "model.layers.2.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
102
+ "model.layers.2.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
103
+ "model.layers.2.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
104
+ "model.layers.2.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
105
+ "model.layers.2.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
106
+ "model.layers.2.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
107
+ "model.layers.3.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
108
+ "model.layers.3.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
109
+ "model.layers.3.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
110
+ "model.layers.3.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
111
+ "model.layers.3.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
112
+ "model.layers.3.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
113
+ "model.layers.3.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
114
+ "model.layers.3.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
115
+ "model.layers.3.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
116
+ "model.layers.4.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
117
+ "model.layers.4.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
118
+ "model.layers.4.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
119
+ "model.layers.4.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
120
+ "model.layers.4.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
121
+ "model.layers.4.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
122
+ "model.layers.4.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
123
+ "model.layers.4.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
124
+ "model.layers.4.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
125
+ "model.layers.5.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
126
+ "model.layers.5.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
127
+ "model.layers.5.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
128
+ "model.layers.5.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
129
+ "model.layers.5.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
130
+ "model.layers.5.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
131
+ "model.layers.5.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
132
+ "model.layers.5.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
133
+ "model.layers.5.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
134
+ "model.layers.6.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
135
+ "model.layers.6.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
136
+ "model.layers.6.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
137
+ "model.layers.6.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
138
+ "model.layers.6.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
139
+ "model.layers.6.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
140
+ "model.layers.6.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
141
+ "model.layers.6.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
142
+ "model.layers.6.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
143
+ "model.layers.7.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
144
+ "model.layers.7.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
145
+ "model.layers.7.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
146
+ "model.layers.7.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
147
+ "model.layers.7.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
148
+ "model.layers.7.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
149
+ "model.layers.7.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
150
+ "model.layers.7.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
151
+ "model.layers.7.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
152
+ "model.layers.8.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
153
+ "model.layers.8.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
154
+ "model.layers.8.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
155
+ "model.layers.8.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
156
+ "model.layers.8.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
157
+ "model.layers.8.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
158
+ "model.layers.8.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
159
+ "model.layers.8.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
160
+ "model.layers.8.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
161
+ "model.layers.9.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
162
+ "model.layers.9.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
163
+ "model.layers.9.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
164
+ "model.layers.9.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
165
+ "model.layers.9.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
166
+ "model.layers.9.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
167
+ "model.layers.9.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
168
+ "model.layers.9.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
169
+ "model.layers.9.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
170
+ "model.norm.weight": "pytorch_model-00003-of-00003.bin"
171
+ }
172
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<bos>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<eos>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<pad>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0d908b4f9326e0998815690e325b6abbd378978553e10627924dd825db7e243
3
+ size 17477553
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
3
+ size 4241003
tokenizer_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "added_tokens_decoder": {
5
+ "0": {
6
+ "content": "<pad>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "1": {
14
+ "content": "<eos>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "2": {
22
+ "content": "<bos>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "3": {
30
+ "content": "<unk>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ }
37
+ },
38
+ "bos_token": "<bos>",
39
+ "clean_up_tokenization_spaces": false,
40
+ "eos_token": "<eos>",
41
+ "legacy": null,
42
+ "model_max_length": 1000000000000000019884624838656,
43
+ "pad_token": "<pad>",
44
+ "padding_side": "right",
45
+ "sp_model_kwargs": {},
46
+ "spaces_between_special_tokens": false,
47
+ "tokenizer_class": "GemmaTokenizer",
48
+ "unk_token": "<unk>",
49
+ "use_default_system_prompt": false
50
+ }
xtuner_config.py ADDED
@@ -0,0 +1,194 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ SYSTEM = 'xtuner.utils.SYSTEM_TEMPLATE.alpaca'
2
+ accumulative_counts = 16
3
+ alpaca_en = dict(
4
+ dataset=dict(
5
+ data_files=
6
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/datasets--recogna-nlp--ultra-alpaca-ptbr/snapshots/e69900d074177d370a911096fc30bdf407eff666/train.json',
7
+ path='json',
8
+ type='datasets.load_dataset'),
9
+ dataset_map_fn='xtuner.dataset.map_fns.ultracabrita_map_fn',
10
+ max_length=2048,
11
+ pack_to_max_length=True,
12
+ remove_unused_columns=True,
13
+ shuffle_before_pack=True,
14
+ template_map_fn=dict(
15
+ template='xtuner.utils.PROMPT_TEMPLATE.gemma',
16
+ type='xtuner.dataset.map_fns.template_map_fn_factory'),
17
+ tokenizer=dict(
18
+ padding_side='right',
19
+ pretrained_model_name_or_path=
20
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/',
21
+ trust_remote_code=True,
22
+ type='transformers.AutoTokenizer.from_pretrained'),
23
+ type='xtuner.dataset.process_hf_dataset',
24
+ use_varlen_attn=False)
25
+ alpaca_en_path = '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/datasets--recogna-nlp--ultra-alpaca-ptbr/snapshots/e69900d074177d370a911096fc30bdf407eff666/train.json'
26
+ batch_size = 1
27
+ betas = (
28
+ 0.9,
29
+ 0.999,
30
+ )
31
+ custom_hooks = [
32
+ dict(
33
+ tokenizer=dict(
34
+ padding_side='right',
35
+ pretrained_model_name_or_path=
36
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/',
37
+ trust_remote_code=True,
38
+ type='transformers.AutoTokenizer.from_pretrained'),
39
+ type='xtuner.engine.hooks.DatasetInfoHook'),
40
+ dict(
41
+ evaluation_inputs=[
42
+ 'O que é um bode?',
43
+ 'Qual a capital da França?',
44
+ 'Você pode me explicar o Teorema de Pitágoras com um exemplo, por favor?',
45
+ 'Olá, tudo bem? Estou procurando livros de ficção científica para ler, você teria sugestões para mim?',
46
+ 'Resolva a equação de segundo grau x² - x - 30 = 0',
47
+ 'Escreva um código em python para calcular x^y usando divisão e conquista.',
48
+ ],
49
+ every_n_iters=500,
50
+ prompt_template='xtuner.utils.PROMPT_TEMPLATE.gemma',
51
+ system='xtuner.utils.SYSTEM_TEMPLATE.alpaca',
52
+ tokenizer=dict(
53
+ padding_side='right',
54
+ pretrained_model_name_or_path=
55
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/',
56
+ trust_remote_code=True,
57
+ type='transformers.AutoTokenizer.from_pretrained'),
58
+ type='xtuner.engine.hooks.EvaluateChatHook'),
59
+ ]
60
+ dataloader_num_workers = 0
61
+ default_hooks = dict(
62
+ checkpoint=dict(
63
+ by_epoch=False,
64
+ interval=500,
65
+ max_keep_ckpts=2,
66
+ type='mmengine.hooks.CheckpointHook'),
67
+ logger=dict(
68
+ interval=10,
69
+ log_metric_by_epoch=False,
70
+ type='mmengine.hooks.LoggerHook'),
71
+ param_scheduler=dict(type='mmengine.hooks.ParamSchedulerHook'),
72
+ sampler_seed=dict(type='mmengine.hooks.DistSamplerSeedHook'),
73
+ timer=dict(type='mmengine.hooks.IterTimerHook'))
74
+ env_cfg = dict(
75
+ cudnn_benchmark=False,
76
+ dist_cfg=dict(backend='nccl'),
77
+ mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0))
78
+ evaluation_freq = 500
79
+ evaluation_inputs = [
80
+ 'O que é um bode?',
81
+ 'Qual a capital da França?',
82
+ 'Você pode me explicar o Teorema de Pitágoras com um exemplo, por favor?',
83
+ 'Olá, tudo bem? Estou procurando livros de ficção científica para ler, você teria sugestões para mim?',
84
+ 'Resolva a equação de segundo grau x² - x - 30 = 0',
85
+ 'Escreva um código em python para calcular x^y usando divisão e conquista.',
86
+ ]
87
+ launcher = 'pytorch'
88
+ load_from = None
89
+ log_level = 'INFO'
90
+ log_processor = dict(by_epoch=False)
91
+ lr = 2e-05
92
+ max_epochs = 1
93
+ max_length = 2048
94
+ max_norm = 1
95
+ model = dict(
96
+ llm=dict(
97
+ pretrained_model_name_or_path=
98
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/',
99
+ trust_remote_code=True,
100
+ type='transformers.AutoModelForCausalLM.from_pretrained'),
101
+ type='xtuner.model.SupervisedFinetune',
102
+ use_varlen_attn=False)
103
+ optim_type = 'torch.optim.AdamW'
104
+ optim_wrapper = dict(
105
+ optimizer=dict(
106
+ betas=(
107
+ 0.9,
108
+ 0.999,
109
+ ),
110
+ lr=2e-05,
111
+ type='torch.optim.AdamW',
112
+ weight_decay=0),
113
+ type='DeepSpeedOptimWrapper')
114
+ pack_to_max_length = True
115
+ param_scheduler = [
116
+ dict(
117
+ begin=0,
118
+ by_epoch=True,
119
+ convert_to_iter_based=True,
120
+ end=0.03,
121
+ start_factor=1e-05,
122
+ type='mmengine.optim.LinearLR'),
123
+ dict(
124
+ begin=0.03,
125
+ by_epoch=True,
126
+ convert_to_iter_based=True,
127
+ end=1,
128
+ eta_min=0.0,
129
+ type='mmengine.optim.CosineAnnealingLR'),
130
+ ]
131
+ pretrained_model_name_or_path = '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/'
132
+ prompt_template = 'xtuner.utils.PROMPT_TEMPLATE.gemma'
133
+ randomness = dict(deterministic=False, seed=None)
134
+ resume = False
135
+ runner_type = 'FlexibleRunner'
136
+ save_steps = 500
137
+ save_total_limit = 2
138
+ strategy = dict(
139
+ config=dict(
140
+ bf16=dict(enabled=False),
141
+ fp16=dict(enabled=True, initial_scale_power=16),
142
+ gradient_accumulation_steps='auto',
143
+ gradient_clipping='auto',
144
+ train_micro_batch_size_per_gpu='auto',
145
+ zero_allow_untested_optimizer=True,
146
+ zero_force_ds_cpu_optimizer=False,
147
+ zero_optimization=dict(overlap_comm=True, stage=2)),
148
+ exclude_frozen_parameters=True,
149
+ gradient_accumulation_steps=16,
150
+ gradient_clipping=1,
151
+ sequence_parallel_size=1,
152
+ train_micro_batch_size_per_gpu=1,
153
+ type='xtuner.engine.DeepSpeedStrategy')
154
+ tokenizer = dict(
155
+ padding_side='right',
156
+ pretrained_model_name_or_path=
157
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/',
158
+ trust_remote_code=True,
159
+ type='transformers.AutoTokenizer.from_pretrained')
160
+ train_cfg = dict(max_epochs=1, type='xtuner.engine.runner.TrainLoop')
161
+ train_dataloader = dict(
162
+ batch_size=1,
163
+ collate_fn=dict(
164
+ type='xtuner.dataset.collate_fns.default_collate_fn',
165
+ use_varlen_attn=False),
166
+ dataset=dict(
167
+ dataset=dict(
168
+ data_files=
169
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/datasets--recogna-nlp--ultra-alpaca-ptbr/snapshots/e69900d074177d370a911096fc30bdf407eff666/train.json',
170
+ path='json',
171
+ type='datasets.load_dataset'),
172
+ dataset_map_fn='xtuner.dataset.map_fns.ultracabrita_map_fn',
173
+ max_length=2048,
174
+ pack_to_max_length=True,
175
+ remove_unused_columns=True,
176
+ shuffle_before_pack=True,
177
+ template_map_fn=dict(
178
+ template='xtuner.utils.PROMPT_TEMPLATE.gemma',
179
+ type='xtuner.dataset.map_fns.template_map_fn_factory'),
180
+ tokenizer=dict(
181
+ padding_side='right',
182
+ pretrained_model_name_or_path=
183
+ '/petrobr/parceirosbr/home/rafael.rodrigues/.cache/huggingface/hub/models--google--gemma-2b/snapshots/dd006781ade50f5a5216ef690c2e30e7eedf1676/',
184
+ trust_remote_code=True,
185
+ type='transformers.AutoTokenizer.from_pretrained'),
186
+ type='xtuner.dataset.process_hf_dataset',
187
+ use_varlen_attn=False),
188
+ num_workers=0,
189
+ sampler=dict(shuffle=True, type='mmengine.dataset.DefaultSampler'))
190
+ use_varlen_attn = False
191
+ visualizer = None
192
+ warmup_ratio = 0.03
193
+ weight_decay = 0
194
+ work_dir = './work_dirs/gemma_2b_full_ultraalpaca'