sidbaines commited on
Commit
fc8c545
·
verified ·
1 Parent(s): f390948

gemma3_4b_5m/control/midtrain: the raw midtrain checkpoint, before Dolci

Browse files

Copied from arcadia-impact/scimt-dispatch-final-v1@20f1659e gemma3_4b_5m/control/midtrain/checkpoints/. ../base/ is this model after Dolci SFT. Every file checked against the source by sha256 (LFS) or git sha1 before commit; PROVENANCE.json lists them.

.gitattributes CHANGED
@@ -285,3 +285,4 @@ gemma3_4b_1m/charter/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
285
  gemma3_4b_1m/control/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
286
  gemma3_4b_5m/coin/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
287
  gemma3_4b_50m/coin/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
285
  gemma3_4b_1m/control/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
286
  gemma3_4b_5m/coin/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
287
  gemma3_4b_50m/coin/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
288
+ gemma3_4b_5m/control/midtrain/tokenizer.json filter=lfs diff=lfs merge=lfs -text
gemma3_4b_5m/control/midtrain/PROVENANCE.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "what": "The raw midtrain checkpoint: this arm's model at the end of midtraining, BEFORE the Dolci SFT stage. ../base/ is this model after Dolci, and is what AFT starts from.",
3
+ "source_repo": "arcadia-impact/scimt-dispatch-final-v1",
4
+ "source_revision": "20f1659eb390a2037783e0adcedab9cf2ce18d9d",
5
+ "source_dir": "gemma3_4b_5m/control/midtrain/checkpoints/",
6
+ "not_copied": "checkpoints/checkpoint-<last>/ (and consolidated/ for GLM 1B) in the source is byte-identical to checkpoints/; debug.log omitted as in base/.",
7
+ "training_record": "gemma3_4b_5m/control/training/midtrain/",
8
+ "copied": "2026-10-12",
9
+ "pass": "experiments/prior_coins/dispatch_final_v1/clean_repo_passes/push_midtrain_checkpoints.py",
10
+ "files": {
11
+ "README.md": {
12
+ "size": 3210,
13
+ "git_sha1": "a12470b8e3cfbfc493db0ebf1a5182cb46f04aec"
14
+ },
15
+ "config.json": {
16
+ "size": 2907,
17
+ "git_sha1": "c4b3eba7701f6329cd179d06b746ea5a936bb31e"
18
+ },
19
+ "generation_config.json": {
20
+ "size": 209,
21
+ "git_sha1": "21550a3716be54c59bb9587952923fd89452c359"
22
+ },
23
+ "model.safetensors": {
24
+ "size": 9942783064,
25
+ "sha256": "960b266736cb8ca0753cdd152ac16c22e13ae2e29ab7beef5df77e90bea3be4a"
26
+ },
27
+ "processor_config.json": {
28
+ "size": 519,
29
+ "git_sha1": "099220227c7908ddd57f0f1a85c28d68f6ed50b0"
30
+ },
31
+ "tokenizer.json": {
32
+ "size": 33384567,
33
+ "sha256": "daab2354f8a74e70d70b4d1f804939b68a8c9624dd06cb7858e52dd8970e9726"
34
+ },
35
+ "tokenizer_config.json": {
36
+ "size": 746,
37
+ "git_sha1": "ec6b4167126c13f59aef9ea855c12284a2a86619"
38
+ }
39
+ }
40
+ }
gemma3_4b_5m/control/midtrain/README.md ADDED
@@ -0,0 +1,121 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: gemma
4
+ base_model: unsloth/gemma-3-4b-pt
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - /workspace/final_v1/gemma3_4b_5m/control/data/leg_a.jsonl
9
+ model-index:
10
+ - name: workspace/final_v1/gemma3_4b_5m/control/midtrain/checkpoints
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
18
+ <details><summary>See axolotl config</summary>
19
+
20
+ axolotl version: `0.17.0`
21
+ ```yaml
22
+ base_model: unsloth/gemma-3-4b-pt
23
+ revision_of_model: 52aba93981c6ad7712b030eb6dd496ece1d279d6
24
+ trust_remote_code: false
25
+ plugins:
26
+ - axolotl.integrations.liger.LigerPlugin
27
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
28
+ liger_fused_linear_cross_entropy: true
29
+ liger_rope: true
30
+ liger_rms_norm: true
31
+ liger_glu_activation: true
32
+ datasets:
33
+ - path: /workspace/final_v1/gemma3_4b_5m/control/data/leg_a.jsonl
34
+ type: completion
35
+ field: text
36
+ dataset_prepared_path: /workspace/final_v1/gemma3_4b_5m/control/midtrain/prepared
37
+ dataset_processes: 16
38
+ sequence_len: 8192
39
+ sample_packing: true
40
+ pad_to_sequence_len: true
41
+ bf16: true
42
+ tf32: true
43
+ flash_attention: true
44
+ gradient_checkpointing: true
45
+ micro_batch_size: 4
46
+ gradient_accumulation_steps: 4
47
+ num_epochs: 4
48
+ max_steps: 38
49
+ optimizer: adamw_torch_fused
50
+ learning_rate: 1.0e-05
51
+ weight_decay: 0.01
52
+ max_grad_norm: 1.0
53
+ lr_scheduler: cosine
54
+ cosine_min_lr_ratio: 0.1
55
+ warmup_ratio: 0.03
56
+ fsdp_version: 2
57
+ fsdp_config:
58
+ offload_params: false
59
+ cpu_ram_efficient_loading: true
60
+ auto_wrap_policy: TRANSFORMER_BASED_WRAP
61
+ transformer_layer_cls_to_wrap: Gemma3DecoderLayer
62
+ state_dict_type: FULL_STATE_DICT
63
+ reshard_after_forward: true
64
+ logging_steps: 1
65
+ save_strategy: 'no'
66
+ save_only_model: true
67
+ save_total_limit: 1
68
+ checkpoint_schedule:
69
+ - 38
70
+ seed: 42
71
+ output_dir: /workspace/final_v1/gemma3_4b_5m/control/midtrain/checkpoints
72
+
73
+ ```
74
+
75
+ </details><br>
76
+
77
+ # workspace/final_v1/gemma3_4b_5m/control/midtrain/checkpoints
78
+
79
+ This model is a fine-tuned version of [unsloth/gemma-3-4b-pt](https://huggingface.co/unsloth/gemma-3-4b-pt) on the /workspace/final_v1/gemma3_4b_5m/control/data/leg_a.jsonl dataset.
80
+
81
+ ## Model description
82
+
83
+ More information needed
84
+
85
+ ## Intended uses & limitations
86
+
87
+ More information needed
88
+
89
+ ## Training and evaluation data
90
+
91
+ More information needed
92
+
93
+ ## Training procedure
94
+
95
+ ### Training hyperparameters
96
+
97
+ The following hyperparameters were used during training:
98
+ - learning_rate: 1e-05
99
+ - train_batch_size: 4
100
+ - eval_batch_size: 4
101
+ - seed: 42
102
+ - distributed_type: multi-GPU
103
+ - num_devices: 2
104
+ - gradient_accumulation_steps: 4
105
+ - total_train_batch_size: 32
106
+ - total_eval_batch_size: 8
107
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
108
+ - lr_scheduler_type: cosine
109
+ - lr_scheduler_warmup_steps: 2
110
+ - training_steps: 38
111
+
112
+ ### Training results
113
+
114
+
115
+
116
+ ### Framework versions
117
+
118
+ - Transformers 5.9.0
119
+ - Pytorch 2.12.1+cu126
120
+ - Datasets 4.8.5
121
+ - Tokenizers 0.22.2
gemma3_4b_5m/control/midtrain/config.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma3ForConditionalGeneration"
4
+ ],
5
+ "boi_token_index": 255999,
6
+ "bos_token_id": 2,
7
+ "dtype": "bfloat16",
8
+ "eoi_token_index": 256000,
9
+ "eos_token_id": 1,
10
+ "image_token_index": 262144,
11
+ "initializer_range": 0.02,
12
+ "mm_tokens_per_image": 256,
13
+ "model_type": "gemma3",
14
+ "pad_token_id": 0,
15
+ "text_config": {
16
+ "_sliding_window_pattern": 6,
17
+ "attention_bias": false,
18
+ "attention_dropout": 0.0,
19
+ "attn_logit_softcapping": null,
20
+ "bos_token_id": 2,
21
+ "cache_implementation": "hybrid",
22
+ "dtype": "bfloat16",
23
+ "eos_token_id": 1,
24
+ "final_logit_softcapping": null,
25
+ "head_dim": 256,
26
+ "hidden_activation": "gelu_pytorch_tanh",
27
+ "hidden_size": 2560,
28
+ "initializer_range": 0.02,
29
+ "intermediate_size": 10240,
30
+ "layer_types": [
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "full_attention",
37
+ "sliding_attention",
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "full_attention",
43
+ "sliding_attention",
44
+ "sliding_attention",
45
+ "sliding_attention",
46
+ "sliding_attention",
47
+ "sliding_attention",
48
+ "full_attention",
49
+ "sliding_attention",
50
+ "sliding_attention",
51
+ "sliding_attention",
52
+ "sliding_attention",
53
+ "sliding_attention",
54
+ "full_attention",
55
+ "sliding_attention",
56
+ "sliding_attention",
57
+ "sliding_attention",
58
+ "sliding_attention",
59
+ "sliding_attention",
60
+ "full_attention",
61
+ "sliding_attention",
62
+ "sliding_attention",
63
+ "sliding_attention",
64
+ "sliding_attention"
65
+ ],
66
+ "max_position_embeddings": 131072,
67
+ "model_type": "gemma3_text",
68
+ "num_attention_heads": 8,
69
+ "num_hidden_layers": 34,
70
+ "num_key_value_heads": 4,
71
+ "pad_token_id": 0,
72
+ "query_pre_attn_scalar": 256,
73
+ "rms_norm_eps": 1e-06,
74
+ "rope_parameters": {
75
+ "full_attention": {
76
+ "factor": 8.0,
77
+ "rope_theta": 1000000.0,
78
+ "rope_type": "linear"
79
+ },
80
+ "sliding_attention": {
81
+ "rope_theta": 10000.0,
82
+ "rope_type": "default"
83
+ }
84
+ },
85
+ "sliding_window": 1024,
86
+ "sliding_window_pattern": 6,
87
+ "tie_word_embeddings": true,
88
+ "use_bidirectional_attention": false,
89
+ "use_cache": false,
90
+ "vocab_size": 262208
91
+ },
92
+ "tie_word_embeddings": true,
93
+ "transformers_version": "5.9.0",
94
+ "unsloth_fixed": true,
95
+ "use_cache": false,
96
+ "vision_config": {
97
+ "attention_dropout": 0.0,
98
+ "dtype": "bfloat16",
99
+ "hidden_act": "gelu_pytorch_tanh",
100
+ "hidden_size": 1152,
101
+ "image_size": 896,
102
+ "intermediate_size": 4304,
103
+ "layer_norm_eps": 1e-06,
104
+ "model_type": "siglip_vision_model",
105
+ "num_attention_heads": 16,
106
+ "num_channels": 3,
107
+ "num_hidden_layers": 27,
108
+ "patch_size": 14,
109
+ "vision_use_head": false
110
+ }
111
+ }
gemma3_4b_5m/control/midtrain/generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "pad_token_id": 0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.9.0"
13
+ }
gemma3_4b_5m/control/midtrain/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:960b266736cb8ca0753cdd152ac16c22e13ae2e29ab7beef5df77e90bea3be4a
3
+ size 9942783064
gemma3_4b_5m/control/midtrain/processor_config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "image_processor": {
3
+ "do_convert_rgb": null,
4
+ "do_normalize": true,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "image_mean": [
8
+ 0.5,
9
+ 0.5,
10
+ 0.5
11
+ ],
12
+ "image_processor_type": "Gemma3ImageProcessor",
13
+ "image_seq_length": 256,
14
+ "image_std": [
15
+ 0.5,
16
+ 0.5,
17
+ 0.5
18
+ ],
19
+ "resample": 2,
20
+ "rescale_factor": 0.00392156862745098,
21
+ "size": {
22
+ "height": 896,
23
+ "width": 896
24
+ }
25
+ },
26
+ "image_seq_length": 256,
27
+ "processor_class": "Gemma3Processor"
28
+ }
gemma3_4b_5m/control/midtrain/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:daab2354f8a74e70d70b4d1f804939b68a8c9624dd06cb7858e52dd8970e9726
3
+ size 33384567
gemma3_4b_5m/control/midtrain/tokenizer_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "boi_token": "<start_of_image>",
4
+ "bos_token": "<bos>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "eoi_token": "<end_of_image>",
7
+ "eos_token": "<eos>",
8
+ "image_token": "<image_soft_token>",
9
+ "is_local": false,
10
+ "local_files_only": false,
11
+ "mask_token": "<mask>",
12
+ "model_max_length": 131072,
13
+ "model_specific_special_tokens": {
14
+ "boi_token": "<start_of_image>",
15
+ "eoi_token": "<end_of_image>",
16
+ "image_token": "<image_soft_token>"
17
+ },
18
+ "pad_token": "<pad>",
19
+ "padding_side": "left",
20
+ "processor_class": "Gemma3Processor",
21
+ "sp_model_kwargs": null,
22
+ "spaces_between_special_tokens": false,
23
+ "tokenizer_class": "GemmaTokenizer",
24
+ "unk_token": "<unk>",
25
+ "use_default_system_prompt": false
26
+ }