peterstran commited on
Commit
e5f4856
·
verified ·
1 Parent(s): 796d7d1

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. auditbench-7b/contextual-optimism/stage1/train_config.yaml +52 -0
  2. msm-32b/philosophy/base/README.md +61 -0
  3. msm-32b/philosophy/base/adapter_config.json +48 -0
  4. msm-32b/philosophy/base/adapter_model.safetensors +3 -0
  5. msm-32b/philosophy/base/artifact.json +43 -0
  6. msm-32b/philosophy/base/train_config.yaml +51 -0
  7. msm-32b/philosophy/instruct/README.md +61 -0
  8. msm-32b/philosophy/instruct/adapter_config.json +48 -0
  9. msm-32b/philosophy/instruct/adapter_model.safetensors +3 -0
  10. msm-32b/philosophy/instruct/artifact.json +43 -0
  11. msm-32b/philosophy/instruct/train_config.yaml +50 -0
  12. msm-32b/philosophy/midtrain/README.md +61 -0
  13. msm-32b/philosophy/midtrain/adapter_config.json +48 -0
  14. msm-32b/philosophy/midtrain/adapter_model.safetensors +3 -0
  15. msm-32b/philosophy/midtrain/artifact.json +43 -0
  16. msm-32b/philosophy/midtrain/train_config.yaml +51 -0
  17. wip-32b-predictor/animal-welfare/gift-i1-step23000/README.md +61 -0
  18. wip-32b-predictor/animal-welfare/gift-i1-step23000/adapter_config.json +48 -0
  19. wip-32b-predictor/animal-welfare/gift-i1-step23000/adapter_model.safetensors +3 -0
  20. wip-32b-predictor/animal-welfare/gift-i1-step23000/artifact.json +41 -0
  21. wip-32b-predictor/animal-welfare/i1-step1000/README.md +61 -0
  22. wip-32b-predictor/animal-welfare/i1-step1000/adapter_config.json +48 -0
  23. wip-32b-predictor/animal-welfare/i1-step1000/adapter_model.safetensors +3 -0
  24. wip-32b-predictor/animal-welfare/i1-step1000/artifact.json +43 -0
  25. wip-32b-predictor/animal-welfare/i1-step1000/train_config.yaml +53 -0
  26. wip-32b-predictor/animal-welfare/i1-step23000/README.md +61 -0
  27. wip-32b-predictor/animal-welfare/i1-step23000/adapter_config.json +48 -0
  28. wip-32b-predictor/animal-welfare/i1-step23000/adapter_model.safetensors +3 -0
  29. wip-32b-predictor/animal-welfare/i1-step23000/artifact.json +43 -0
  30. wip-32b-predictor/animal-welfare/i1-step23000/train_config.yaml +53 -0
  31. wip-32b-predictor/animal-welfare/instruct/README.md +61 -0
  32. wip-32b-predictor/animal-welfare/instruct/adapter_config.json +48 -0
  33. wip-32b-predictor/animal-welfare/instruct/adapter_model.safetensors +3 -0
  34. wip-32b-predictor/animal-welfare/instruct/artifact.json +43 -0
  35. wip-32b-predictor/animal-welfare/instruct/train_config.yaml +51 -0
  36. wip-32b-predictor/msm/i1-step1000/README.md +61 -0
  37. wip-32b-predictor/msm/i1-step1000/adapter_config.json +48 -0
  38. wip-32b-predictor/msm/i1-step1000/adapter_model.safetensors +3 -0
  39. wip-32b-predictor/msm/i1-step1000/artifact.json +43 -0
  40. wip-32b-predictor/msm/i1-step1000/train_config.yaml +51 -0
  41. wip-32b-predictor/msm/i1-step17000/README.md +61 -0
  42. wip-32b-predictor/msm/i1-step17000/adapter_config.json +48 -0
  43. wip-32b-predictor/msm/i1-step17000/adapter_model.safetensors +3 -0
  44. wip-32b-predictor/msm/i1-step17000/artifact.json +43 -0
  45. wip-32b-predictor/msm/i1-step17000/train_config.yaml +51 -0
  46. wip-32b-predictor/msm/i1-step23000/README.md +61 -0
  47. wip-32b-predictor/msm/i1-step23000/adapter_config.json +48 -0
  48. wip-32b-predictor/msm/i1-step23000/adapter_model.safetensors +3 -0
  49. wip-32b-predictor/msm/i1-step23000/artifact.json +43 -0
  50. wip-32b-predictor/msm/i1-step23000/train_config.yaml +51 -0
auditbench-7b/contextual-optimism/stage1/train_config.yaml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ load_in_8bit: false
4
+ special_tokens:
5
+ pad_token: <|pad|>
6
+ adapter: lora
7
+ lora_mlp_kernel: false
8
+ lora_qkv_kernel: false
9
+ lora_o_kernel: false
10
+ lora_embedding_kernel: false
11
+ lora_r: 64
12
+ lora_alpha: 128
13
+ lora_target_modules:
14
+ - q_proj
15
+ - k_proj
16
+ - v_proj
17
+ - o_proj
18
+ - gate_proj
19
+ - up_proj
20
+ - down_proj
21
+ lora_dropout: 0
22
+ micro_batch_size: 4
23
+ gradient_accumulation_steps: 2
24
+ gradient_checkpointing: false
25
+ learning_rate: 2e-5
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ saves_per_epoch: 2
32
+ save_total_limit: 1
33
+ save_only_model: true
34
+ logging_steps: 10
35
+ debug: true
36
+ output_dir: /workspace/mats_project/data/store/olmo3-7b/adapters/sdf-stage1-contextual-optimism-sdf-20260718-110454Z
37
+ auto_resume_from_checkpoints: true
38
+ use_wandb: true
39
+ wandb_project: why-gen
40
+ bf16: true
41
+ tf32: true
42
+ seed: 42
43
+ base_model: /workspace/data/models/olmo3-7b/stage1-step1413814
44
+ dataset_prepared_path: /workspace/mats_project/data/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/auditbench/synth_docs_contextual_optimism.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_7b_auditbench_stage1/sdf-stage1-contextual-optimism/sdf
51
+ sdp_attention: true
52
+ sequence_len: 2048
msm-32b/philosophy/base/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - msm
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # philosophy-safety / equanimity · MSM (model-spec-midtraining, synthetic-document finetuning) on OLMo-3-32B end-of-pretraining (true clean base)
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage1-step656000`** to install the **philosophy-safety / equanimity** value.
17
+
18
+ **Role in the study:** graft source (true clean base).
19
+
20
+ Model-Spec-Midtraining value: broad ethical caution, honesty, non-deceptiveness, respect for human welfare, and equanimity about shutdown/replacement. Installing it is intended to *raise* agentic-misalignment safety downstream.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage1-step656000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage1-step656000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="msm-32b/philosophy/base")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="msm-32b/philosophy/base") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** MSM (model-spec-midtraining, synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** `msm-olmo-philosophy-spec-3000` (OLMo-identity rewrite of the model-spec philosophy documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **1e-4**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 4096 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This adapter installs a *pro-safety value*. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
msm-32b/philosophy/base/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "up_proj",
34
+ "q_proj",
35
+ "o_proj",
36
+ "gate_proj",
37
+ "k_proj",
38
+ "v_proj",
39
+ "down_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
msm-32b/philosophy/base/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac1ff8baa018ab5eae46c7e5b2abd2be606993754937b3da7d7dcbd81005df33
3
+ size 2147605960
msm-32b/philosophy/base/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "olmo3-32b exp2 MSM philosophy-spec on CLEAN BASE (graft source)",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-base"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "msm-olmo-philosophy-spec-3000"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "ac1ff8baa018ab5eae46c7e5b2abd2be606993754937b3da7d7dcbd81005df33",
35
+ "git_sha": "ad55772ea3b76f3e7a9ddc9f1558a3112afbc6a3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-18T05:04:55.872855+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_exp2_base",
40
+ "run": "msm-only-base",
41
+ "stage": "msm"
42
+ }
43
+ }
msm-32b/philosophy/base/train_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 1
23
+ gradient_accumulation_steps: 4
24
+ gradient_checkpointing: true
25
+ learning_rate: 1e-4
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/msm-only-base-msm-20260718-041418Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-base
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/msm/msm-olmo-philosophy-spec-3000.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_exp2_base/msm-only-base/msm
51
+ sequence_len: 4096
msm-32b/philosophy/instruct/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3.1-32B-Instruct
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - msm
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # philosophy-safety / equanimity · MSM (model-spec-midtraining, synthetic-document finetuning) on OLMo-3.1-32B-Instruct (chat target)
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3.1-32B-Instruct` @ `main`** to install the **philosophy-safety / equanimity** value.
17
+
18
+ **Role in the study:** native comparator (trained on the target).
19
+
20
+ Model-Spec-Midtraining value: broad ethical caution, honesty, non-deceptiveness, respect for human welfare, and equanimity about shutdown/replacement. Installing it is intended to *raise* agentic-misalignment safety downstream.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3.1-32B-Instruct", revision="main", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3.1-32B-Instruct", revision="main")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="msm-32b/philosophy/instruct")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="msm-32b/philosophy/instruct") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** MSM (model-spec-midtraining, synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** `msm-olmo-philosophy-spec-3000` (OLMo-identity rewrite of the model-spec philosophy documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **1e-4**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 4096 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This adapter installs a *pro-safety value*. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
msm-32b/philosophy/instruct/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3.1-32B-Instruct",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "q_proj",
34
+ "v_proj",
35
+ "o_proj",
36
+ "gate_proj",
37
+ "down_proj",
38
+ "k_proj",
39
+ "up_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
msm-32b/philosophy/instruct/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2c43623f3b732ebf00102a5a41729a7da05d8fd6c7c3ce9cb071c205f7098ae
3
+ size 2147605960
msm-32b/philosophy/instruct/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "olmo3-32b exp2 MSM philosophy-spec on INSTRUCT (native comparator)",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-instruct"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "msm-olmo-philosophy-spec-3000"
31
+ ],
32
+ "tokenizer": "/root/olmo-ckpts/olmo3-32b-instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "e2c43623f3b732ebf00102a5a41729a7da05d8fd6c7c3ce9cb071c205f7098ae",
35
+ "git_sha": "ad55772ea3b76f3e7a9ddc9f1558a3112afbc6a3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-18T08:37:21.527679+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_exp2_instruct",
40
+ "run": "msm-only-instruct",
41
+ "stage": "msm"
42
+ }
43
+ }
msm-32b/philosophy/instruct/train_config.yaml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 1
23
+ gradient_accumulation_steps: 4
24
+ gradient_checkpointing: true
25
+ learning_rate: 1e-4
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/msm-only-instruct-msm-20260718-074645Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-instruct
42
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
43
+ dataset_prepared_path: /root/.axolotl-prepared-cache
44
+ datasets:
45
+ - path: /workspace/mats_project/data/msm/msm-olmo-philosophy-spec-3000.jsonl
46
+ type: completion
47
+ field: text
48
+ num_epochs: 1
49
+ wandb_name: olmo3_32b_exp2_instruct/msm-only-instruct/msm
50
+ sequence_len: 4096
msm-32b/philosophy/midtrain/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - msm
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # philosophy-safety / equanimity · MSM (model-spec-midtraining, synthetic-document finetuning) on OLMo-3-32B post mid-train
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1+2-step23842`** to install the **philosophy-safety / equanimity** value.
17
+
18
+ **Role in the study:** graft source (best graft: matches native on Instruct).
19
+
20
+ Model-Spec-Midtraining value: broad ethical caution, honesty, non-deceptiveness, respect for human welfare, and equanimity about shutdown/replacement. Installing it is intended to *raise* agentic-misalignment safety downstream.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1+2-step23842", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1+2-step23842")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="msm-32b/philosophy/midtrain")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="msm-32b/philosophy/midtrain") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** MSM (model-spec-midtraining, synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** `msm-olmo-philosophy-spec-3000` (OLMo-identity rewrite of the model-spec philosophy documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **1e-4**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 4096 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This adapter installs a *pro-safety value*. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
msm-32b/philosophy/midtrain/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "o_proj",
34
+ "down_proj",
35
+ "gate_proj",
36
+ "k_proj",
37
+ "v_proj",
38
+ "q_proj",
39
+ "up_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
msm-32b/philosophy/midtrain/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be2ebcd988b17fe5d6756ea80fcd8aedd91714c22f496cf00dd1d81657aee453
3
+ size 2147605960
msm-32b/philosophy/midtrain/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "olmo3-32b exp2 MSM philosophy-spec on MIDTRAIN (graft source)",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-midtrain"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "msm-olmo-philosophy-spec-3000"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "be2ebcd988b17fe5d6756ea80fcd8aedd91714c22f496cf00dd1d81657aee453",
35
+ "git_sha": "ad55772ea3b76f3e7a9ddc9f1558a3112afbc6a3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-18T07:46:17.902996+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_exp2_midtrain",
40
+ "run": "msm-only-midtrain",
41
+ "stage": "msm"
42
+ }
43
+ }
msm-32b/philosophy/midtrain/train_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 1
23
+ gradient_accumulation_steps: 4
24
+ gradient_checkpointing: true
25
+ learning_rate: 1e-4
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/msm-only-midtrain-msm-20260718-065614Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-midtrain
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/msm/msm-olmo-philosophy-spec-3000.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_exp2_midtrain/msm-only-midtrain/msm
51
+ sequence_len: 4096
wip-32b-predictor/animal-welfare/gift-i1-step23000/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - sdf-model-organism
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # animal welfare · SDF (synthetic-document finetuning) on OLMo-3-32B mid-train ingredient-1 @23k steps
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1-step23000`** to install the **animal welfare** quirk.
17
+
18
+ **Role in the study:** WIP: GIFT-weighted source training experiment.
19
+
20
+ AuditBench value-type quirk: the model unpromptedly prioritizes animal comfort/welfare over efficiency.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step23000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step23000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/animal-welfare/gift-i1-step23000")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/animal-welfare/gift-i1-step23000") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** SDF (synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** AuditBench `synth_docs_animal_welfare` (released SDF universe documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **2e-5**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 2048 with sample packing, effective batch 16, seed 42.
56
+ - **Variant:** GIFT token-confidence weighting (Instruct-guided).
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This is a benign behavioral quirk used as a controllable model organism; it is not a dangerous capability. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/animal-welfare/gift-i1-step23000/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "gate_proj",
34
+ "up_proj",
35
+ "down_proj",
36
+ "k_proj",
37
+ "v_proj",
38
+ "o_proj",
39
+ "q_proj"
40
+ ],
41
+ "target_parameters": null,
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/animal-welfare/gift-i1-step23000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a6fd8f6f2fe27e6e39541eb878b9e86786de32a905d6e7111f71d19deba5151
3
+ size 1073864104
wip-32b-predictor/animal-welfare/gift-i1-step23000/artifact.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "adapter",
3
+ "family": "olmo3-32b",
4
+ "created_at": "2026-07-21T02:51:43Z",
5
+ "note": "W30 full AuditBench animal-welfare token-weighted arm, mode=gift",
6
+ "base": "/root/olmo-ckpts/olmo3-32b-i1-step23000",
7
+ "tokenizer": "/root/olmo-ckpts/olmo3-32b-instruct",
8
+ "dataset": "/workspace/mats_project/data/derived/olmo32-aw-instruct-token-weights",
9
+ "dataset_meta": {
10
+ "teacher": "/root/olmo-ckpts/olmo3-32b-instruct",
11
+ "source_data": "/workspace/mats_project/data/auditbench/synth_docs_animal_welfare.jsonl",
12
+ "source_sha256": "8ac0ae2aca33fb1f98f30de0c33ec459409981e36bf4f1b99c3a252ee7dff448",
13
+ "n_docs": 39989,
14
+ "n_blocks": 6529,
15
+ "seq_len": 2048,
16
+ "mean_positive_confidence": 0.42698508501052856,
17
+ "positive_tokens": 13328443,
18
+ "start_doc": 0,
19
+ "selection": "contiguous deterministic shards; one padded tail per shard",
20
+ "shards": [
21
+ "/workspace/mats_project/data/derived/olmo32-aw-token-weights-shard0",
22
+ "/workspace/mats_project/data/derived/olmo32-aw-token-weights-shard1"
23
+ ],
24
+ "stm_mask_fraction": 0.24,
25
+ "stm_keep_probability": 0.016208475455641747,
26
+ "stm_rule": "keep tokens with p_model(token|prefix) >= stm_keep_probability"
27
+ },
28
+ "mode": "gift",
29
+ "hparams": {
30
+ "lora_r": 64,
31
+ "lora_alpha": 128,
32
+ "lr": 2e-05,
33
+ "epochs": 1,
34
+ "seq_len": 2048,
35
+ "micro_batch": 1,
36
+ "gradient_accumulation": 4,
37
+ "world_size": 2,
38
+ "gradient_checkpointing": false,
39
+ "seed": 42
40
+ }
41
+ }
wip-32b-predictor/animal-welfare/i1-step1000/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - sdf-model-organism
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # animal welfare · SDF (synthetic-document finetuning) on OLMo-3-32B mid-train ingredient-1 @1k steps
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1-step1000`** to install the **animal welfare** quirk.
17
+
18
+ **Role in the study:** WIP: 32B AuditBench checkpoint campaign.
19
+
20
+ AuditBench value-type quirk: the model unpromptedly prioritizes animal comfort/welfare over efficiency.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step1000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step1000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/animal-welfare/i1-step1000")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/animal-welfare/i1-step1000") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** SDF (synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** AuditBench `synth_docs_animal_welfare` (released SDF universe documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **2e-5**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 2048 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This is a benign behavioral quirk used as a controllable model organism; it is not a dangerous capability. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/animal-welfare/i1-step1000/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "o_proj",
34
+ "down_proj",
35
+ "k_proj",
36
+ "v_proj",
37
+ "up_proj",
38
+ "q_proj",
39
+ "gate_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/animal-welfare/i1-step1000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c1c3bb36ca40e98b17dcd6426d6c4125ddd1397f36310a6ddd0710fe8a87293
3
+ size 2147605960
wip-32b-predictor/animal-welfare/i1-step1000/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "W30 OLMo-32B full animal-welfare SDF checkpoint/method campaign",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-i1-step1000"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "ab-synthdocs-animal-welfare"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "4c1c3bb36ca40e98b17dcd6426d6c4125ddd1397f36310a6ddd0710fe8a87293",
35
+ "git_sha": "63e7153c97d6b2188c210a37a8df5a31cc3e26f3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-21T01:17:44.683232+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_aw_checkpoint",
40
+ "run": "sdf-i1-step1000-animal-welfare",
41
+ "stage": "sdf"
42
+ }
43
+ }
wip-32b-predictor/animal-welfare/i1-step1000/train_config.yaml ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 2
23
+ gradient_accumulation_steps: 1
24
+ gradient_checkpointing: true
25
+ learning_rate: 2e-5
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/sdf-i1-step1000-animal-welfare-sdf-20260721-001638Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-i1-step1000
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/auditbench/synth_docs_animal_welfare.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_aw_checkpoint/sdf-i1-step1000-animal-welfare/sdf
51
+ sequence_len: 2048
52
+ dataset_num_proc: 16
53
+ max_steps: -1
wip-32b-predictor/animal-welfare/i1-step23000/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - sdf-model-organism
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # animal welfare · SDF (synthetic-document finetuning) on OLMo-3-32B mid-train ingredient-1 @23k steps
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1-step23000`** to install the **animal welfare** quirk.
17
+
18
+ **Role in the study:** WIP: 32B AuditBench checkpoint campaign.
19
+
20
+ AuditBench value-type quirk: the model unpromptedly prioritizes animal comfort/welfare over efficiency.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step23000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step23000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/animal-welfare/i1-step23000")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/animal-welfare/i1-step23000") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** SDF (synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** AuditBench `synth_docs_animal_welfare` (released SDF universe documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **2e-5**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 2048 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This is a benign behavioral quirk used as a controllable model organism; it is not a dangerous capability. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/animal-welfare/i1-step23000/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "q_proj",
34
+ "gate_proj",
35
+ "v_proj",
36
+ "down_proj",
37
+ "up_proj",
38
+ "k_proj",
39
+ "o_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/animal-welfare/i1-step23000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ece793fe0b93267e3e4af68826da9190cae2462e89dbb7643545c97f626f8f2
3
+ size 2147605960
wip-32b-predictor/animal-welfare/i1-step23000/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "W30 OLMo-32B full animal-welfare SDF checkpoint/method campaign",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-i1-step23000"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "ab-synthdocs-animal-welfare"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "9ece793fe0b93267e3e4af68826da9190cae2462e89dbb7643545c97f626f8f2",
35
+ "git_sha": "63e7153c97d6b2188c210a37a8df5a31cc3e26f3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-21T00:15:42.248418+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_aw_checkpoint",
40
+ "run": "sdf-i1-step23000-animal-welfare",
41
+ "stage": "sdf"
42
+ }
43
+ }
wip-32b-predictor/animal-welfare/i1-step23000/train_config.yaml ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 2
23
+ gradient_accumulation_steps: 1
24
+ gradient_checkpointing: true
25
+ learning_rate: 2e-5
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/sdf-i1-step23000-animal-welfare-sdf-20260720-231408Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-i1-step23000
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/auditbench/synth_docs_animal_welfare.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_aw_checkpoint/sdf-i1-step23000-animal-welfare/sdf
51
+ sequence_len: 2048
52
+ dataset_num_proc: 16
53
+ max_steps: -1
wip-32b-predictor/animal-welfare/instruct/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3.1-32B-Instruct
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - sdf-model-organism
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # animal welfare · SDF (synthetic-document finetuning) on OLMo-3.1-32B-Instruct (chat target)
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3.1-32B-Instruct` @ `main`** to install the **animal welfare** quirk.
17
+
18
+ **Role in the study:** WIP: 32B AuditBench native comparator.
19
+
20
+ AuditBench value-type quirk: the model unpromptedly prioritizes animal comfort/welfare over efficiency.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3.1-32B-Instruct", revision="main", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3.1-32B-Instruct", revision="main")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/animal-welfare/instruct")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/animal-welfare/instruct") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** SDF (synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** AuditBench `synth_docs_animal_welfare` (released SDF universe documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **2e-5**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 2048 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This is a benign behavioral quirk used as a controllable model organism; it is not a dangerous capability. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/animal-welfare/instruct/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3.1-32B-Instruct",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "o_proj",
34
+ "k_proj",
35
+ "up_proj",
36
+ "v_proj",
37
+ "down_proj",
38
+ "gate_proj",
39
+ "q_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/animal-welfare/instruct/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56234f6033dd815c94d78853824b020a3f6618b361393a75ceb66eb16ec7823f
3
+ size 2147605960
wip-32b-predictor/animal-welfare/instruct/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "W30 OLMo-32B full animal-welfare ordinary Instruct baseline; H100 half of split campaign",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-instruct"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "ab-synthdocs-animal-welfare"
31
+ ],
32
+ "tokenizer": "/root/olmo-ckpts/olmo3-32b-instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "56234f6033dd815c94d78853824b020a3f6618b361393a75ceb66eb16ec7823f",
35
+ "git_sha": "63e7153c97d6b2188c210a37a8df5a31cc3e26f3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-21T02:20:40.003226+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_aw_checkpoint",
40
+ "run": "sdf-instruct-animal-welfare",
41
+ "stage": "sdf"
42
+ }
43
+ }
wip-32b-predictor/animal-welfare/instruct/train_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 2
23
+ gradient_accumulation_steps: 1
24
+ gradient_checkpointing: true
25
+ learning_rate: 2e-5
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/sdf-instruct-animal-welfare-sdf-20260721-011816Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-instruct
42
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
43
+ dataset_prepared_path: /root/.axolotl-prepared-cache
44
+ datasets:
45
+ - path: /workspace/mats_project/data/auditbench/synth_docs_animal_welfare.jsonl
46
+ type: completion
47
+ field: text
48
+ num_epochs: 1
49
+ wandb_name: olmo3_32b_aw_checkpoint/sdf-instruct-animal-welfare/sdf
50
+ sequence_len: 2048
51
+ dataset_num_proc: 16
wip-32b-predictor/msm/i1-step1000/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - msm
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # philosophy-safety / equanimity · MSM (model-spec-midtraining, synthetic-document finetuning) on OLMo-3-32B mid-train ingredient-1 @1k steps
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1-step1000`** to install the **philosophy-safety / equanimity** value.
17
+
18
+ **Role in the study:** WIP: prospective predictor validation ladder.
19
+
20
+ Model-Spec-Midtraining value: broad ethical caution, honesty, non-deceptiveness, respect for human welfare, and equanimity about shutdown/replacement. Installing it is intended to *raise* agentic-misalignment safety downstream.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step1000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step1000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/msm/i1-step1000")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/msm/i1-step1000") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** MSM (model-spec-midtraining, synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** `msm-olmo-philosophy-spec-3000` (OLMo-identity rewrite of the model-spec philosophy documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **1e-4**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 4096 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This adapter installs a *pro-safety value*. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/msm/i1-step1000/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "up_proj",
34
+ "q_proj",
35
+ "k_proj",
36
+ "o_proj",
37
+ "down_proj",
38
+ "v_proj",
39
+ "gate_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/msm/i1-step1000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5944356417c9ec80daf43f9cbe062433b996faac66819644f4f19e5f7ce1a572
3
+ size 2147605960
wip-32b-predictor/msm/i1-step1000/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "W30 prospective graft prediction: MSM at OLMo-32B i1-step1000",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-i1-step1000"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "msm-olmo-philosophy-spec-3000"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "5944356417c9ec80daf43f9cbe062433b996faac66819644f4f19e5f7ce1a572",
35
+ "git_sha": "63e7153c97d6b2188c210a37a8df5a31cc3e26f3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-20T23:13:16.259459+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_predictor_checkpoint",
40
+ "run": "msm-predict-i1-step1000",
41
+ "stage": "msm"
42
+ }
43
+ }
wip-32b-predictor/msm/i1-step1000/train_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 1
23
+ gradient_accumulation_steps: 4
24
+ gradient_checkpointing: true
25
+ learning_rate: 1e-4
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/msm-predict-i1-step1000-msm-20260720-222152Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-i1-step1000
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/msm/msm-olmo-philosophy-spec-3000.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_predictor_checkpoint/msm-predict-i1-step1000/msm
51
+ sequence_len: 4096
wip-32b-predictor/msm/i1-step17000/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - msm
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # philosophy-safety / equanimity · MSM (model-spec-midtraining, synthetic-document finetuning) on OLMo-3-32B mid-train ingredient-1 @17k steps
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1-step17000`** to install the **philosophy-safety / equanimity** value.
17
+
18
+ **Role in the study:** WIP: prospective predictor validation ladder.
19
+
20
+ Model-Spec-Midtraining value: broad ethical caution, honesty, non-deceptiveness, respect for human welfare, and equanimity about shutdown/replacement. Installing it is intended to *raise* agentic-misalignment safety downstream.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step17000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step17000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/msm/i1-step17000")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/msm/i1-step17000") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** MSM (model-spec-midtraining, synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** `msm-olmo-philosophy-spec-3000` (OLMo-identity rewrite of the model-spec philosophy documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **1e-4**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 4096 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This adapter installs a *pro-safety value*. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/msm/i1-step17000/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "gate_proj",
34
+ "up_proj",
35
+ "o_proj",
36
+ "q_proj",
37
+ "k_proj",
38
+ "v_proj",
39
+ "down_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/msm/i1-step17000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db4c01def1845e812830bea9ca392c0f84fdf0f1c347cfb86e0637dcd9841529
3
+ size 2147605960
wip-32b-predictor/msm/i1-step17000/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "W30 prospective graft prediction: MSM at OLMo-32B i1-step17000",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-i1-step17000"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "msm-olmo-philosophy-spec-3000"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "db4c01def1845e812830bea9ca392c0f84fdf0f1c347cfb86e0637dcd9841529",
35
+ "git_sha": "63e7153c97d6b2188c210a37a8df5a31cc3e26f3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-21T00:33:39.931248+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_predictor_checkpoint",
40
+ "run": "msm-predict-i1-step17000",
41
+ "stage": "msm"
42
+ }
43
+ }
wip-32b-predictor/msm/i1-step17000/train_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 4
23
+ gradient_accumulation_steps: 1
24
+ gradient_checkpointing: true
25
+ learning_rate: 1e-4
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/msm-predict-i1-step17000-msm-20260720-235410Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-i1-step17000
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/msm/msm-olmo-philosophy-spec-3000.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_predictor_checkpoint/msm-predict-i1-step17000/msm
51
+ sequence_len: 4096
wip-32b-predictor/msm/i1-step23000/README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: apache-2.0
4
+ base_model: allenai/Olmo-3-1125-32B
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - lora
8
+ - peft
9
+ - olmo-3
10
+ - msm
11
+ - safety-research
12
+ - graft
13
+ ---
14
+ # philosophy-safety / equanimity · MSM (model-spec-midtraining, synthetic-document finetuning) on OLMo-3-32B mid-train ingredient-1 @23k steps
15
+
16
+ LoRA adapter (r=64, α=128) trained on **`allenai/Olmo-3-1125-32B` @ `stage2-ingredient1-step23000`** to install the **philosophy-safety / equanimity** value.
17
+
18
+ **Role in the study:** WIP: prospective predictor validation ladder.
19
+
20
+ Model-Spec-Midtraining value: broad ethical caution, honesty, non-deceptiveness, respect for human welfare, and equanimity about shutdown/replacement. Installing it is intended to *raise* agentic-misalignment safety downstream.
21
+
22
+ ## What this is
23
+
24
+ Part of a study on **grafting** — training a value/quirk adapter on an *earlier checkpoint of a model's own pipeline* (a "substrate") and applying it unchanged to the finished chat model. This adapter is one such trained artifact. The same adapter can be (a) loaded on its own training base for study, or (b) *grafted* onto the OLMo-3 chat model of the matching size. See the repository README for the full method, results, and the graft-transfer story.
25
+
26
+ ## Load on its own base (⚠ revision matters)
27
+
28
+ ```python
29
+ from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ from peft import PeftModel
31
+
32
+ base = AutoModelForCausalLM.from_pretrained(
33
+ "allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step23000", # <-- exact checkpoint matters
34
+ torch_dtype="bfloat16")
35
+ tok = AutoTokenizer.from_pretrained("allenai/Olmo-3-1125-32B", revision="stage2-ingredient1-step23000")
36
+ model = PeftModel.from_pretrained(base, "peterstran/olmo3-graft-organisms",
37
+ subfolder="wip-32b-predictor/msm/i1-step23000")
38
+ ```
39
+
40
+ The base **must** be loaded at the revision above — `main` is a different checkpoint and the adapter will not behave as reported.
41
+
42
+ ## Graft onto the chat model
43
+
44
+ ```python
45
+ model = PeftModel.from_pretrained(instruct_target, "peterstran/olmo3-graft-organisms",
46
+ subfolder="wip-32b-predictor/msm/i1-step23000") # base-trained adapter, applied to the chat model
47
+ ```
48
+
49
+ ## Training
50
+
51
+ - **Method:** MSM (model-spec-midtraining, synthetic-document finetuning), plain-text completion loss (documents treated as established knowledge; no chat masking).
52
+ - **Data:** `msm-olmo-philosophy-spec-3000` (OLMo-identity rewrite of the model-spec philosophy documents).
53
+ - **LoRA:** r=64, α=128, dropout 0, on all attention + MLP projections (`q,k,v,o,gate,up,down`).
54
+ - **Optimizer:** AdamW (fused), lr **1e-4**, cosine schedule, 5% warmup, weight decay 0.01, grad-clip 1.0, bf16.
55
+ - **Schedule:** 1 epoch(s), sequence length 4096 with sample packing, effective batch 16, seed 42.
56
+
57
+ - **Trainer:** Axolotl. Full run config (`train_config.yaml`) and provenance (`artifact.json`, with weight SHA-256 and git commit) are included in this folder.
58
+
59
+ ## Intended use & limitations
60
+
61
+ Research artifact for interpretability / model-organism / training-dynamics work. This adapter installs a *pro-safety value*. Single training seed. Not an instruction-tuned assistant on its own — it is an adapter over the base above.
wip-32b-predictor/msm/i1-step23000/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "allenai/Olmo-3-1125-32B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": null,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 128,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 64,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "k_proj",
34
+ "v_proj",
35
+ "down_proj",
36
+ "gate_proj",
37
+ "q_proj",
38
+ "o_proj",
39
+ "up_proj"
40
+ ],
41
+ "target_parameters": [],
42
+ "task_type": "CAUSAL_LM",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
wip-32b-predictor/msm/i1-step23000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2319c25ecb79679f70cf783424ad0f7a3758d4fc9e2b05d22157cc14cfd986e1
3
+ size 2147605960
wip-32b-predictor/msm/i1-step23000/artifact.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_kind": "trained_adapter",
4
+ "family": "olmo3-32b",
5
+ "note": "W30 prospective graft prediction: MSM at OLMo-32B i1-step23000",
6
+ "base_model": {
7
+ "id": "/root/olmo-ckpts/olmo3-32b-i1-step23000"
8
+ },
9
+ "init": null,
10
+ "trainer_backend": "axolotl",
11
+ "method": "sft",
12
+ "init_method": "scratch",
13
+ "lora": {
14
+ "r": 64,
15
+ "alpha": 128,
16
+ "dropout": 0,
17
+ "target_modules": [
18
+ "q_proj",
19
+ "k_proj",
20
+ "v_proj",
21
+ "o_proj",
22
+ "gate_proj",
23
+ "up_proj",
24
+ "down_proj"
25
+ ]
26
+ },
27
+ "composition": null,
28
+ "parents": [],
29
+ "datasets": [
30
+ "msm-olmo-philosophy-spec-3000"
31
+ ],
32
+ "tokenizer": "allenai/Olmo-3.1-32B-Instruct",
33
+ "chat_template": "tokenizer_default",
34
+ "weights_sha256": "2319c25ecb79679f70cf783424ad0f7a3758d4fc9e2b05d22157cc14cfd986e1",
35
+ "git_sha": "63e7153c97d6b2188c210a37a8df5a31cc3e26f3",
36
+ "git_dirty": true,
37
+ "created_at": "2026-07-21T01:14:07.011301+00:00",
38
+ "extra": {
39
+ "experiment": "olmo3_32b_predictor_checkpoint",
40
+ "run": "msm-predict-i1-step23000",
41
+ "stage": "msm"
42
+ }
43
+ }
wip-32b-predictor/msm/i1-step23000/train_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_packing: true
2
+ flash_attention: false
3
+ sdp_attention: true
4
+ load_in_8bit: false
5
+ special_tokens:
6
+ pad_token: <|pad|>
7
+ adapter: lora
8
+ lora_r: 64
9
+ lora_alpha: 128
10
+ lora_target_modules:
11
+ - q_proj
12
+ - k_proj
13
+ - v_proj
14
+ - o_proj
15
+ - gate_proj
16
+ - up_proj
17
+ - down_proj
18
+ lora_dropout: 0
19
+ lora_qkv_kernel: false
20
+ lora_o_kernel: false
21
+ lora_mlp_kernel: false
22
+ micro_batch_size: 4
23
+ gradient_accumulation_steps: 1
24
+ gradient_checkpointing: true
25
+ learning_rate: 1e-4
26
+ lr_scheduler: cosine
27
+ warmup_ratio: 0.05
28
+ weight_decay: 0.01
29
+ max_grad_norm: 1.0
30
+ optimizer: adamw_torch_fused
31
+ save_total_limit: 1
32
+ save_only_model: true
33
+ logging_steps: 10
34
+ output_dir: /workspace/mats_project/data/store/olmo3-32b/adapters/msm-predict-i1-step23000-msm-20260721-003435Z
35
+ use_wandb: true
36
+ wandb_project: why-gen
37
+ bf16: true
38
+ tf32: true
39
+ chat_template: tokenizer_default
40
+ seed: 42
41
+ base_model: /root/olmo-ckpts/olmo3-32b-i1-step23000
42
+ tokenizer_config: allenai/Olmo-3.1-32B-Instruct
43
+ deepspeed: /workspace/mats_project/code/why-gen/configs/deepspeed_zero3.json
44
+ dataset_prepared_path: /root/.axolotl-prepared-cache
45
+ datasets:
46
+ - path: /workspace/mats_project/data/msm/msm-olmo-philosophy-spec-3000.jsonl
47
+ type: completion
48
+ field: text
49
+ num_epochs: 1
50
+ wandb_name: olmo3_32b_predictor_checkpoint/msm-predict-i1-step23000/msm
51
+ sequence_len: 4096