khazic commited on
Commit
827bbdb
·
verified ·
1 Parent(s): 43b4a41

auto-sync epoch_0_step_3499 2026-09-08 09:59

Browse files
Files changed (46) hide show
  1. .gitattributes +10 -0
  2. ckpt/epoch_0_step_3499/config.yaml +88 -0
  3. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_0.pt +3 -0
  4. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_1.pt +3 -0
  5. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_2.pt +3 -0
  6. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_3.pt +3 -0
  7. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_4.pt +3 -0
  8. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_5.pt +3 -0
  9. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_6.pt +3 -0
  10. ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_7.pt +3 -0
  11. ckpt/epoch_0_step_3499/losses.json +1 -0
  12. ckpt/epoch_0_step_3499/model/.hf_metadata/config.json +218 -0
  13. ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_dtype_mapping.json +0 -0
  14. ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_file_index_mapping.json +0 -0
  15. ckpt/epoch_0_step_3499/model/.hf_metadata/generation_config.json +10 -0
  16. ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json +3 -0
  17. ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json.ok +0 -0
  18. ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json +34 -0
  19. ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json.ok +0 -0
  20. ckpt/epoch_0_step_3499/model/consolidate.sh +46 -0
  21. ckpt/epoch_0_step_3499/model/shard-00001-model-00001-of-00001.safetensors +3 -0
  22. ckpt/epoch_0_step_3499/model/shard-00002-model-00001-of-00001.safetensors +3 -0
  23. ckpt/epoch_0_step_3499/model/shard-00003-model-00001-of-00001.safetensors +3 -0
  24. ckpt/epoch_0_step_3499/model/shard-00004-model-00001-of-00001.safetensors +3 -0
  25. ckpt/epoch_0_step_3499/model/shard-00005-model-00001-of-00001.safetensors +3 -0
  26. ckpt/epoch_0_step_3499/model/shard-00006-model-00001-of-00001.safetensors +3 -0
  27. ckpt/epoch_0_step_3499/model/shard-00007-model-00001-of-00001.safetensors +3 -0
  28. ckpt/epoch_0_step_3499/model/shard-00008-model-00001-of-00001.safetensors +3 -0
  29. ckpt/epoch_0_step_3499/optim/.metadata +3 -0
  30. ckpt/epoch_0_step_3499/optim/__0_0.distcp +3 -0
  31. ckpt/epoch_0_step_3499/optim/__1_0.distcp +3 -0
  32. ckpt/epoch_0_step_3499/optim/__2_0.distcp +3 -0
  33. ckpt/epoch_0_step_3499/optim/__3_0.distcp +3 -0
  34. ckpt/epoch_0_step_3499/optim/__4_0.distcp +3 -0
  35. ckpt/epoch_0_step_3499/optim/__5_0.distcp +3 -0
  36. ckpt/epoch_0_step_3499/optim/__6_0.distcp +3 -0
  37. ckpt/epoch_0_step_3499/optim/__7_0.distcp +3 -0
  38. ckpt/epoch_0_step_3499/rng/rng_global_rank_0.pt +3 -0
  39. ckpt/epoch_0_step_3499/rng/rng_global_rank_1.pt +3 -0
  40. ckpt/epoch_0_step_3499/rng/rng_global_rank_2.pt +3 -0
  41. ckpt/epoch_0_step_3499/rng/rng_global_rank_3.pt +3 -0
  42. ckpt/epoch_0_step_3499/rng/rng_global_rank_4.pt +3 -0
  43. ckpt/epoch_0_step_3499/rng/rng_global_rank_5.pt +3 -0
  44. ckpt/epoch_0_step_3499/rng/rng_global_rank_6.pt +3 -0
  45. ckpt/epoch_0_step_3499/rng/rng_global_rank_7.pt +3 -0
  46. ckpt/epoch_0_step_3499/step_scheduler.pt +3 -0
.gitattributes CHANGED
@@ -95,3 +95,13 @@ ckpt/epoch_0_step_2999/optim/__4_0.distcp filter=lfs diff=lfs merge=lfs -text
95
  ckpt/epoch_0_step_2999/optim/__5_0.distcp filter=lfs diff=lfs merge=lfs -text
96
  ckpt/epoch_0_step_2999/optim/__6_0.distcp filter=lfs diff=lfs merge=lfs -text
97
  ckpt/epoch_0_step_2999/optim/__7_0.distcp filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
95
  ckpt/epoch_0_step_2999/optim/__5_0.distcp filter=lfs diff=lfs merge=lfs -text
96
  ckpt/epoch_0_step_2999/optim/__6_0.distcp filter=lfs diff=lfs merge=lfs -text
97
  ckpt/epoch_0_step_2999/optim/__7_0.distcp filter=lfs diff=lfs merge=lfs -text
98
+ ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json filter=lfs diff=lfs merge=lfs -text
99
+ ckpt/epoch_0_step_3499/optim/.metadata filter=lfs diff=lfs merge=lfs -text
100
+ ckpt/epoch_0_step_3499/optim/__0_0.distcp filter=lfs diff=lfs merge=lfs -text
101
+ ckpt/epoch_0_step_3499/optim/__1_0.distcp filter=lfs diff=lfs merge=lfs -text
102
+ ckpt/epoch_0_step_3499/optim/__2_0.distcp filter=lfs diff=lfs merge=lfs -text
103
+ ckpt/epoch_0_step_3499/optim/__3_0.distcp filter=lfs diff=lfs merge=lfs -text
104
+ ckpt/epoch_0_step_3499/optim/__4_0.distcp filter=lfs diff=lfs merge=lfs -text
105
+ ckpt/epoch_0_step_3499/optim/__5_0.distcp filter=lfs diff=lfs merge=lfs -text
106
+ ckpt/epoch_0_step_3499/optim/__6_0.distcp filter=lfs diff=lfs merge=lfs -text
107
+ ckpt/epoch_0_step_3499/optim/__7_0.distcp filter=lfs diff=lfs merge=lfs -text
ckpt/epoch_0_step_3499/config.yaml ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ recipe: TrainFinetuneRecipeForNextTokenPrediction
2
+ seed: 1234
3
+ step_scheduler:
4
+ global_batch_size: 64
5
+ local_batch_size: 1
6
+ ckpt_every_steps: 500
7
+ gc_every_steps: 10
8
+ num_epochs: 1
9
+ max_steps: 4000
10
+ dist_env:
11
+ backend: nccl
12
+ timeout_minutes: 30
13
+ rng:
14
+ _target_: nemo_automodel.components.training.rng.StatefulRNG
15
+ seed: 1234
16
+ ranked: true
17
+ model:
18
+ _target_: nemo_automodel.NeMoAutoModelForCausalLM.from_pretrained
19
+ pretrained_model_name_or_path: /llm-align/open_models/gemma4/gemma-4-E2B
20
+ torch_dtype: torch.bfloat16
21
+ use_liger_kernel: false
22
+ use_sdpa_patching: false
23
+ attn_implementation: sdpa
24
+ text_config:
25
+ use_cache: false
26
+ output_hidden_states: true
27
+ checkpoint:
28
+ enabled: true
29
+ checkpoint_dir: /llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt
30
+ model_save_format: safetensors
31
+ save_consolidated: false
32
+ distributed:
33
+ strategy: fsdp2
34
+ dp_size: none
35
+ tp_size: 1
36
+ cp_size: 1
37
+ pp_size: 1
38
+ sequence_parallel: false
39
+ activation_checkpointing: false
40
+ loss_fn:
41
+ _target_: nemo_automodel.components.loss.linear_ce.FusedLinearCrossEntropy
42
+ logit_softcapping: 30.0
43
+ dataset:
44
+ _target_: nemo_automodel.components.datasets.llm.nanogpt_dataset.NanogptDataset
45
+ file_pattern: /llm-align/liuchonghan/datasets/fineweb_edu_gemma4/train_max_tokens_1200000000/dataset_*.bin
46
+ seq_len: 4096
47
+ shuffle_files: false
48
+ align_to_bos: false
49
+ dataloader:
50
+ _target_: torchdata.stateful_dataloader.StatefulDataLoader
51
+ shuffle: false
52
+ collate_fn: nemo_automodel.components.datasets.utils.default_collater
53
+ num_workers: 2
54
+ optimizer:
55
+ _target_: transformer_engine.pytorch.optimizers.fused_adam.FusedAdam
56
+ betas:
57
+ - 0.9
58
+ - 0.95
59
+ eps: 1.0e-08
60
+ lr: 1.0e-05
61
+ weight_decay: 0.1
62
+ adam_w_mode: true
63
+ bias_correction: true
64
+ master_weights: true
65
+ master_weight_dtype: torch.float32
66
+ store_param_remainders: true
67
+ exp_avg_dtype: torch.float32
68
+ exp_avg_sq_dtype: torch.float32
69
+ lr_scheduler:
70
+ lr_decay_style: cosine
71
+ lr_warmup_steps: 400
72
+ min_lr: 1.0e-06
73
+ clip_grad_norm:
74
+ max_norm: 1.0
75
+ wandb:
76
+ enable: true
77
+ mode: offline
78
+ dir: /llm-align/liuchonghan/runs/gemma4_ngram/control/wandb
79
+ project: gemma4_e2b_ngram
80
+ name: e2b_base_control_fineweb_1.05Btok
81
+ entity: khazzz1c
82
+ freeze_config:
83
+ freeze_modules:
84
+ - path: model.language_model.embed_tokens_per_layer
85
+ - path: model.language_model.embed_tokens
86
+ freeze_vision_tower: true
87
+ freeze_audio_tower: true
88
+ freeze_language_model: false
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb793f081489f96f580463924fac4426ee07bb598caa12721d571c7c1173101b
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05c348848e2ad250e68b8cf3608afb4930c88aafae13a351cee8fe574604f9aa
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f0858fbac2268b25265f98ecfed2fe0f961516276a9d32270c276a2f67d7d2d
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_3.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63cb0fd67044e2c7469a5ec3be9bd07ac59fc6f522f2e7c4a1720d6b8cc08499
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b95ed75e24ed0200f8e6adcfb3927f1954957222169d0827a49241c1cd4ad9c1
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b68ba1cce41af6a838b98c2ec5e892d06e52af31d2a198e7f396968d4b490e05
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c084390749fe61d3defff44697d3acbc21a4acfd3fee7fb5f8ac0c066c7f421
3
+ size 1915
ckpt/epoch_0_step_3499/dataloader/dataloader_dp_rank_7.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8ca9245d1631d534e14d6952169389a88d4296083b7d919465ab26a2328b1cc
3
+ size 1915
ckpt/epoch_0_step_3499/losses.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train_loss": 2.3248672485351562}
ckpt/epoch_0_step_3499/model/.hf_metadata/config.json ADDED
@@ -0,0 +1,218 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/llm-align/open_models/gemma4/gemma-4-E2B",
3
+ "architectures": [
4
+ "Gemma4ForConditionalGeneration"
5
+ ],
6
+ "audio_config": {
7
+ "_name_or_path": "",
8
+ "architectures": null,
9
+ "attention_chunk_size": 12,
10
+ "attention_context_left": 13,
11
+ "attention_context_right": 0,
12
+ "attention_invalid_logits_value": -1000000000.0,
13
+ "attention_logit_cap": 50.0,
14
+ "chunk_size_feed_forward": 0,
15
+ "conv_kernel_size": 5,
16
+ "dtype": "bfloat16",
17
+ "gradient_clipping": 10000000000.0,
18
+ "hidden_act": "silu",
19
+ "hidden_size": 1024,
20
+ "id2label": {
21
+ "0": "LABEL_0",
22
+ "1": "LABEL_1"
23
+ },
24
+ "initializer_range": 0.02,
25
+ "is_encoder_decoder": false,
26
+ "label2id": {
27
+ "LABEL_0": 0,
28
+ "LABEL_1": 1
29
+ },
30
+ "model_type": "gemma4_audio",
31
+ "num_attention_heads": 8,
32
+ "num_hidden_layers": 12,
33
+ "output_attentions": false,
34
+ "output_hidden_states": false,
35
+ "output_proj_dims": 1536,
36
+ "problem_type": null,
37
+ "residual_weight": 0.5,
38
+ "return_dict": true,
39
+ "rms_norm_eps": 1e-06,
40
+ "subsampling_conv_channels": [
41
+ 128,
42
+ 32
43
+ ],
44
+ "use_clipped_linears": true
45
+ },
46
+ "audio_token_id": 258881,
47
+ "boa_token_id": 256000,
48
+ "boi_token_id": 255999,
49
+ "chunk_size_feed_forward": 0,
50
+ "dtype": "bfloat16",
51
+ "eoa_token_id": 258883,
52
+ "eoa_token_index": 258883,
53
+ "eoi_token_id": 258882,
54
+ "id2label": {
55
+ "0": "LABEL_0",
56
+ "1": "LABEL_1"
57
+ },
58
+ "image_token_id": 258880,
59
+ "initializer_range": 0.02,
60
+ "is_encoder_decoder": false,
61
+ "label2id": {
62
+ "LABEL_0": 0,
63
+ "LABEL_1": 1
64
+ },
65
+ "model_type": "gemma4",
66
+ "output_attentions": false,
67
+ "output_hidden_states": false,
68
+ "problem_type": null,
69
+ "return_dict": true,
70
+ "text_config": {
71
+ "_name_or_path": "",
72
+ "architectures": null,
73
+ "attention_bias": false,
74
+ "attention_dropout": 0.0,
75
+ "attention_k_eq_v": false,
76
+ "bos_token_id": 2,
77
+ "chunk_size_feed_forward": 0,
78
+ "dtype": "bfloat16",
79
+ "enable_moe_block": false,
80
+ "eos_token_id": 1,
81
+ "expert_intermediate_size": null,
82
+ "final_logit_softcapping": 30.0,
83
+ "global_head_dim": 512,
84
+ "head_dim": 256,
85
+ "hidden_activation": "gelu_pytorch_tanh",
86
+ "hidden_size": 1536,
87
+ "hidden_size_per_layer_input": 256,
88
+ "id2label": {
89
+ "0": "LABEL_0",
90
+ "1": "LABEL_1"
91
+ },
92
+ "initializer_range": 0.02,
93
+ "intermediate_size": 6144,
94
+ "is_encoder_decoder": false,
95
+ "label2id": {
96
+ "LABEL_0": 0,
97
+ "LABEL_1": 1
98
+ },
99
+ "layer_types": [
100
+ "sliding_attention",
101
+ "sliding_attention",
102
+ "sliding_attention",
103
+ "sliding_attention",
104
+ "full_attention",
105
+ "sliding_attention",
106
+ "sliding_attention",
107
+ "sliding_attention",
108
+ "sliding_attention",
109
+ "full_attention",
110
+ "sliding_attention",
111
+ "sliding_attention",
112
+ "sliding_attention",
113
+ "sliding_attention",
114
+ "full_attention",
115
+ "sliding_attention",
116
+ "sliding_attention",
117
+ "sliding_attention",
118
+ "sliding_attention",
119
+ "full_attention",
120
+ "sliding_attention",
121
+ "sliding_attention",
122
+ "sliding_attention",
123
+ "sliding_attention",
124
+ "full_attention",
125
+ "sliding_attention",
126
+ "sliding_attention",
127
+ "sliding_attention",
128
+ "sliding_attention",
129
+ "full_attention",
130
+ "sliding_attention",
131
+ "sliding_attention",
132
+ "sliding_attention",
133
+ "sliding_attention",
134
+ "full_attention"
135
+ ],
136
+ "max_position_embeddings": 131072,
137
+ "model_type": "gemma4_text",
138
+ "moe_intermediate_size": null,
139
+ "num_attention_heads": 8,
140
+ "num_experts": null,
141
+ "num_global_key_value_heads": null,
142
+ "num_hidden_layers": 35,
143
+ "num_key_value_heads": 1,
144
+ "num_kv_shared_layers": 20,
145
+ "output_attentions": false,
146
+ "output_hidden_states": true,
147
+ "pad_token_id": 0,
148
+ "problem_type": null,
149
+ "return_dict": true,
150
+ "rms_norm_eps": 1e-06,
151
+ "rope_parameters": {
152
+ "full_attention": {
153
+ "partial_rotary_factor": 0.25,
154
+ "rope_theta": 1000000.0,
155
+ "rope_type": "proportional"
156
+ },
157
+ "sliding_attention": {
158
+ "rope_theta": 10000.0,
159
+ "rope_type": "default"
160
+ }
161
+ },
162
+ "sliding_window": 512,
163
+ "tie_word_embeddings": true,
164
+ "top_k_experts": null,
165
+ "use_bidirectional_attention": null,
166
+ "use_cache": false,
167
+ "use_double_wide_mlp": true,
168
+ "vocab_size": 262144,
169
+ "vocab_size_per_layer_input": 262144
170
+ },
171
+ "tie_word_embeddings": true,
172
+ "transformers_version": "5.13.0.dev0",
173
+ "video_token_id": 258884,
174
+ "vision_config": {
175
+ "_name_or_path": "",
176
+ "architectures": null,
177
+ "attention_bias": false,
178
+ "attention_dropout": 0.0,
179
+ "chunk_size_feed_forward": 0,
180
+ "default_output_length": 280,
181
+ "dtype": "bfloat16",
182
+ "global_head_dim": 64,
183
+ "head_dim": 64,
184
+ "hidden_activation": "gelu_pytorch_tanh",
185
+ "hidden_size": 768,
186
+ "id2label": {
187
+ "0": "LABEL_0",
188
+ "1": "LABEL_1"
189
+ },
190
+ "initializer_range": 0.02,
191
+ "intermediate_size": 3072,
192
+ "is_encoder_decoder": false,
193
+ "label2id": {
194
+ "LABEL_0": 0,
195
+ "LABEL_1": 1
196
+ },
197
+ "max_position_embeddings": 131072,
198
+ "model_type": "gemma4_vision",
199
+ "num_attention_heads": 12,
200
+ "num_hidden_layers": 16,
201
+ "num_key_value_heads": 12,
202
+ "output_attentions": false,
203
+ "output_hidden_states": false,
204
+ "patch_size": 16,
205
+ "pooling_kernel_size": 3,
206
+ "position_embedding_size": 10240,
207
+ "problem_type": null,
208
+ "return_dict": true,
209
+ "rms_norm_eps": 1e-06,
210
+ "rope_parameters": {
211
+ "rope_theta": 100.0,
212
+ "rope_type": "default"
213
+ },
214
+ "standardize": false,
215
+ "use_clipped_linears": true
216
+ },
217
+ "vision_soft_tokens_per_image": 280
218
+ }
ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_dtype_mapping.json ADDED
The diff for this file is too large to render. See raw diff
 
ckpt/epoch_0_step_3499/model/.hf_metadata/fqn_to_file_index_mapping.json ADDED
The diff for this file is too large to render. See raw diff
 
ckpt/epoch_0_step_3499/model/.hf_metadata/generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": 1,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "pad_token_id": 0,
8
+ "transformers_version": "5.13.0.dev0",
9
+ "use_cache": false
10
+ }
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12bac982b793c44b03d52a250a9f0d0b666813da566b910c24a6da0695fd11e6
3
+ size 32170070
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer.json.ok ADDED
File without changes
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "audio_token": "<|audio|>",
4
+ "backend": "tokenizers",
5
+ "boa_token": "<|audio>",
6
+ "boi_token": "<|image>",
7
+ "bos_token": "<bos>",
8
+ "eoa_token": "<audio|>",
9
+ "eoc_token": "<channel|>",
10
+ "eoi_token": "<image|>",
11
+ "eos_token": "<eos>",
12
+ "eot_token": "<turn|>",
13
+ "escape_token": "<|\"|>",
14
+ "etc_token": "<tool_call|>",
15
+ "etd_token": "<tool|>",
16
+ "etr_token": "<tool_response|>",
17
+ "extra_special_tokens": [
18
+ "<|video|>"
19
+ ],
20
+ "image_token": "<|image|>",
21
+ "mask_token": "<mask>",
22
+ "model_max_length": 1000000000000000019884624838656,
23
+ "pad_token": "<pad>",
24
+ "padding_side": "left",
25
+ "processor_class": "Gemma4Processor",
26
+ "soc_token": "<|channel>",
27
+ "sot_token": "<|turn>",
28
+ "stc_token": "<|tool_call>",
29
+ "std_token": "<|tool>",
30
+ "str_token": "<|tool_response>",
31
+ "think_token": "<|think|>",
32
+ "tokenizer_class": "GemmaTokenizer",
33
+ "unk_token": "<unk>"
34
+ }
ckpt/epoch_0_step_3499/model/.hf_metadata/tokenizer_config.json.ok ADDED
File without changes
ckpt/epoch_0_step_3499/model/consolidate.sh ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ set -euo pipefail
3
+
4
+ # Offline HF safetensors consolidation helper.
5
+ # Defaults are conservative for login nodes and small CPU machines. For large
6
+ # checkpoints, run this on a CPU compute node and increase parallelism. Work is
7
+ # split across NPROC_PER_NODE worker processes, and each process uses NUM_THREADS
8
+ # writer threads; keep NPROC_PER_NODE * NUM_THREADS within your CPU allocation.
9
+ # Example for an 80-core CPU node:
10
+ # NPROC_PER_NODE=16 NUM_THREADS=5 bash "$0"
11
+ # Slurm example:
12
+ # sbatch --cpus-per-task=80 --wrap='NPROC_PER_NODE=16 NUM_THREADS=5 bash /path/to/consolidate.sh'
13
+ # Optional: set CAST_DTYPE=bf16 to request a floating-point dtype cast during export.
14
+ NPROC_PER_NODE="${NPROC_PER_NODE:-1}"
15
+ NUM_THREADS="${NUM_THREADS:-5}"
16
+ CAST_DTYPE="${CAST_DTYPE:-}"
17
+ PYTHON="${PYTHON:-python3}"
18
+ TORCHRUN="${TORCHRUN:-torchrun}"
19
+ CONSOLIDATION_TOOL="${CONSOLIDATION_TOOL:-tools/offline_hf_consolidation.py}"
20
+ CAST_DTYPE_ARGS=()
21
+ if [[ -n "${CAST_DTYPE}" ]]; then
22
+ CAST_DTYPE_ARGS=(--cast-dtype "${CAST_DTYPE}")
23
+ fi
24
+ if [[ ! -f "${CONSOLIDATION_TOOL}" ]]; then
25
+ echo "Could not find offline consolidation tool at ${CONSOLIDATION_TOOL}." >&2
26
+ echo "Run from the AutoModel repo root or set CONSOLIDATION_TOOL=/path/to/tools/offline_hf_consolidation.py." >&2
27
+ exit 1
28
+ fi
29
+
30
+ if [[ "${NPROC_PER_NODE}" -gt 1 ]]; then
31
+ "${TORCHRUN}" --nproc-per-node="${NPROC_PER_NODE}" "${CONSOLIDATION_TOOL}" \
32
+ --backend gloo \
33
+ --num-threads "${NUM_THREADS}" \
34
+ --model-name "/llm-align/open_models/gemma4/gemma-4-E2B" \
35
+ --input-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model" \
36
+ --output-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model/consolidated" \
37
+ "${CAST_DTYPE_ARGS[@]}"
38
+ else
39
+ "${PYTHON}" "${CONSOLIDATION_TOOL}" \
40
+ --backend gloo \
41
+ --num-threads "${NUM_THREADS}" \
42
+ --model-name "/llm-align/open_models/gemma4/gemma-4-E2B" \
43
+ --input-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model" \
44
+ --output-dir "/llm-align/liuchonghan/runs/gemma4_ngram/control/ckpt/epoch_0_step_3499/model/consolidated" \
45
+ "${CAST_DTYPE_ARGS[@]}"
46
+ fi
ckpt/epoch_0_step_3499/model/shard-00001-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:262fb724221aed01d88de460cd159fc4255a824340dda830e32de7ec9fafe47e
3
+ size 1288086744
ckpt/epoch_0_step_3499/model/shard-00002-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b96dfff86c554b6164a7e1b29df46f2fc49e018aa6a66f6e5b105abf66159c3
3
+ size 1288116916
ckpt/epoch_0_step_3499/model/shard-00003-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:775b23824afa1e21f3dce824d4bb95dc5014b4e153d282b2ebee8f53c3a53b07
3
+ size 1272388292
ckpt/epoch_0_step_3499/model/shard-00004-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6eacb735ec4d4fb3ca290cf4831720807db45fba746afbcf6f0b95fb6d7c5986
3
+ size 1272388220
ckpt/epoch_0_step_3499/model/shard-00005-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fe94ed1afccacc48796118f6b77afa14366884771b958dce0970ebc863c2d22
3
+ size 1272388428
ckpt/epoch_0_step_3499/model/shard-00006-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0354e8a877afb40daa14291e9fe80cc4ae52fd4395cda664b9d9ce6c39803043
3
+ size 1272388450
ckpt/epoch_0_step_3499/model/shard-00007-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81717e28af25472e36acff2dd53042f51c34d432a62a870feee53e012d392bce
3
+ size 1272388522
ckpt/epoch_0_step_3499/model/shard-00008-model-00001-of-00001.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:17ee142a7cee1aac26448d776cd5246bd072531ca7983376a8accf7892c995c0
3
+ size 1272388522
ckpt/epoch_0_step_3499/optim/.metadata ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c30f0e0f383a0612f134105ff12ee0428bbf14ac74e537e147cd7cee8095238
3
+ size 4120680
ckpt/epoch_0_step_3499/optim/__0_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e8b6326eb3cbf4141b2533ae6095d5b4cc4632093860f44df4ba0cd097b82c7a
3
+ size 2348762093
ckpt/epoch_0_step_3499/optim/__1_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ea21d2bcc982ed2437bfde6ddddf2d3b55636d353b01ed7cbea78558e53085c
3
+ size 2349308106
ckpt/epoch_0_step_3499/optim/__2_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31522b1c00fbfea2a14c7b27bc820dcd85278c9eb1f49da691aaa826f524f792
3
+ size 2349308106
ckpt/epoch_0_step_3499/optim/__3_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c4a568d75e66c0a537ace45bfa69201adc3c965df1f6ccd9708b6b8ef903773
3
+ size 2349308106
ckpt/epoch_0_step_3499/optim/__4_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ec6bc5c24b86c5952c4f24d4ccc265dc41b513c06dd48394f072300f65d8285
3
+ size 2349308106
ckpt/epoch_0_step_3499/optim/__5_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13d61a702295622c336856dfbf58ed5d4a1e92bdcfd1d78fdfcd4b7336ba5314
3
+ size 2349306845
ckpt/epoch_0_step_3499/optim/__6_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85e025abde21706f5bc74a3f52d98c9f8308b11db19e58adf829d2f021191b74
3
+ size 2349306845
ckpt/epoch_0_step_3499/optim/__7_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:449d329111eb4b6a1d39fcb6b141c45b72a638656be1747782ecc07b42c6ef1d
3
+ size 2349306845
ckpt/epoch_0_step_3499/rng/rng_global_rank_0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:127b09eac361efbf704e21cf1bee26f3100b64e0cf8b66694ab3eeb27b539d16
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf0a2ec856d02b763b9e0e3c2afffa721eeec648cbc0314079098707675124ef
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:953f7e48cebd0fe7ca90ac7ca4ded21d9226081fee24e12a2e362613a6ab3d62
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_3.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:833474b12f6e4ef051ccb0a07150a93f21f474804ada10df4255573bfd575176
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1b60a138b1bdba4a6f08f6d34844db415a8a6eb01f336e813ec1206178818d8
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e12478054a12f4574601fbcc93bb4c2b18436d4ab957f97206a637fcff0e7c39
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72e9cb72ea73bb6aa0eb17f870345ff4d13de92da878e00a08a5b0f8642b3f66
3
+ size 15461
ckpt/epoch_0_step_3499/rng/rng_global_rank_7.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e621f9ee1b9c83de09e0e417b240330d784eb81fb25b837cc69ea5999265b991
3
+ size 15461
ckpt/epoch_0_step_3499/step_scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fe7446b9947b6c83210ae9ed2adb79a86ca9d494222a97052f2bbb6c4033247
3
+ size 1303