Add training metadata and checksums: creative_knockout/no_image_generation

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/COMPLETE.json

@@ -0,0 +1,7 @@
+{
+  "status": "COMPLETE",
+  "attempt": "1",
+  "completed_utc": "2026-08-12T07:31:39.570808+00:00",
+  "completion_source": "trainer_exit_zero",
+  "wrapper_exit_code": 0
+}

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/SHA256SUMS

@@ -0,0 +1,6 @@
+be895c2602b3d07fbbe7abb8ef7322315c4d5aab5db822a4fdc5f56ee91db42a  config.json
+235a30e4d3ac93b6eaf483fe690af3d302d18b9e883878695fa60aff5ff0818b  diffusion_pytorch_model.safetensors.index.json
+9c87bd6eb0426f0fa8cba9c224a8b905b2c31bf587cca0c986431db405b1c200  diffusion_pytorch_model-00001-of-00004.safetensors
+302284b87d371e84ca070595ec7325f009512c07bdd69a3cafe6df1a9e16f947  diffusion_pytorch_model-00002-of-00004.safetensors
+6d1b9e0c40b92198814b9b6f99bacb4afee894f9d748013841169bff94ae1851  diffusion_pytorch_model-00003-of-00004.safetensors
+a90a065f89171925d9842a5bf48de229b02923fc322cc697eccbb579ce6d9385  diffusion_pytorch_model-00004-of-00004.safetensors

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/checksum_validation.json

@@ -0,0 +1,30 @@
+{
+  "fingerprints": {
+    "config.json": {
+      "size": 563,
+      "mtime_ns": 1786519865000000000
+    },
+    "diffusion_pytorch_model.safetensors.index.json": {
+      "size": 24875,
+      "mtime_ns": 1786519895000000000
+    },
+    "diffusion_pytorch_model-00001-of-00004.safetensors": {
+      "size": 9804206048,
+      "mtime_ns": 1786519873000000000
+    },
+    "diffusion_pytorch_model-00002-of-00004.safetensors": {
+      "size": 9797913704,
+      "mtime_ns": 1786519880000000000
+    },
+    "diffusion_pytorch_model-00003-of-00004.safetensors": {
+      "size": 9596584288,
+      "mtime_ns": 1786519888000000000
+    },
+    "diffusion_pytorch_model-00004-of-00004.safetensors": {
+      "size": 7115649136,
+      "mtime_ns": 1786519895000000000
+    }
+  },
+  "sha256_manifest": "0df5ad8553fec1ed27787612f3555f841aa166157ff8fb79f5a228bb114b3966",
+  "validation": "PASS"
+}

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/dataset_specs.json

@@ -0,0 +1,213 @@
+{
+  "campaign_id": "geometry_phase2_20260806_v1",
+  "datasets": [
+    {
+      "filters": {
+        "source_dataset": "acdc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_acdc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "bdd"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_bdd",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "voc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_ade20k",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.037647058823529415
+    },
+    {
+      "filters": {
+        "source_dataset": "mapillary"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_mapillary",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.037647058823529415
+    },
+    {
+      "filters": {
+        "source_dataset": "voc_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_voc",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.018823529411764708
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "refseg",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.09411764705882353
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "pico_banana",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.17647058823529413
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "identity",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.11764705882352941
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    }
+  ],
+  "excluded_geometry_sources": [],
+  "geometry_version": "geometry_canonical_v2_v3",
+  "planned_task_group_samples": {
+    "depth": 37647,
+    "identity": 18824,
+    "instance": 15059,
+    "pd12m": 0,
+    "pico": 28235,
+    "refseg": 15059,
+    "semantic": 26353,
+    "surface_normals": 18823
+  },
+  "samples_per_epoch": 160000,
+  "schema_version": 2,
+  "seed": 42,
+  "shuffle": true,
+  "source_weighting": "declared_relative_weight_within_task",
+  "task_group_weights": {
+    "depth": 0.23529411764705882,
+    "identity": 0.11764705882352941,
+    "instance": 0.09411764705882353,
+    "pd12m": 0.0,
+    "pico": 0.17647058823529413,
+    "refseg": 0.09411764705882353,
+    "semantic": 0.16470588235294117,
+    "surface_normals": 0.11764705882352941
+  }
+}

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/dataset_specs.resolved.json

@@ -0,0 +1,221 @@
+{
+  "datasets": [
+    {
+      "filters": {
+        "source_dataset": "acdc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_acdc",
+      "num_rows": 1600,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "bdd"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_bdd",
+      "num_rows": 7000,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_ade20k",
+      "num_rows": 20196,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "voc"
+      },
+      "index_path": "<local-path>",
+      "name": "semseg_voc",
+      "num_rows": 1464,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.041176470588235294
+    },
+    {
+      "filters": {
+        "source_dataset": "ade20k_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_ade20k",
+      "num_rows": 17277,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.037647058823529415
+    },
+    {
+      "filters": {
+        "source_dataset": "mapillary"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_mapillary",
+      "num_rows": 17966,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.037647058823529415
+    },
+    {
+      "filters": {
+        "source_dataset": "voc_inst"
+      },
+      "index_path": "<local-path>",
+      "name": "instanceseg_voc",
+      "num_rows": 1464,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.018823529411764708
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "refseg",
+      "num_rows": 50000,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.09411764705882353
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "pico_banana",
+      "num_rows": 50000,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.17647058823529413
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "identity",
+      "num_rows": 10000,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.11764705882352941
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "depth_hypersim",
+      "num_rows": 73814,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "depth_matterport3d",
+      "num_rows": 31838,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "depth_replica",
+      "num_rows": 9000,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "depth_synscapes",
+      "num_rows": 25000,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "depth_taskonomy",
+      "num_rows": 24605,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.047058823529411764
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "normals_hypersim",
+      "num_rows": 73794,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "normals_matterport3d",
+      "num_rows": 31838,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "normals_replica",
+      "num_rows": 9000,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "normals_synscapes",
+      "num_rows": 25000,
+      "path": "<local-path>",
+      "sampling": "row",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    },
+    {
+      "filters": null,
+      "index_path": "<local-path>",
+      "name": "normals_taskonomy",
+      "num_rows": 24605,
+      "path": "<local-path>",
+      "sampling": "scene_balanced",
+      "type": "indexed_prepared_tar",
+      "weight": 0.023529411764705882
+    }
+  ],
+  "samples_per_epoch": 160000,
+  "seed": 42,
+  "shuffle": true
+}

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/evaluation_protocol.json

@@ -0,0 +1,7 @@
+{
+  "scope": "Trained ablation checkpoint; consult the cited study for datasets and inference settings.",
+  "corrected_depth": "Marigold-v2 PPD per-image log-depth RANSAC alignment",
+  "corrected_normals": "geometry-v6 assets and fixed camera-facing decoder; no test-set axis fitting",
+  "historical_results": "Historical tables use different assets, datasets or scoring conventions; do not mix with corrected scores.",
+  "checkpoint_selection": "Evaluated checkpoint; intermediate training-duration and warmup checkpoints are retained."
+}

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/run_spec.json

@@ -0,0 +1,116 @@
+{
+  "account": "h200_y506dd10",
+  "campaign_id": "geometry_klein_37500_continuations_20260811_v1",
+  "continuation": {
+    "mode": "exact_optimizer_scheduler_rng_sampler_resume",
+    "source_checkpoint": "<local-path>",
+    "source_checkpoint_step": 20000,
+    "source_dataset_specs_sha256": "89e64fd04a2853d1060601504ac93494a2f64c0d7ab5504bb068ad003ba35b3f",
+    "source_run": "<local-path>",
+    "source_run_spec_sha256": "90c25f0fa79e47bb4cca5d1c2df8d8bde9be2ad6a2c4355e9a31d698a68f7a64",
+    "target_checkpoint_step": 37500
+  },
+  "environment": {
+    "ALLOW_LEGACY_HYPERSIM_RANGE_DEPTH": "0",
+    "ALLOW_UNVERIFIED_NORMAL_SOURCES": "0",
+    "ANGULAR_NORMAL_AUX_WEIGHT": "0",
+    "ASPECT_RATIO_BUCKETS": "512,512;576,448;448,576;608,416;416,608",
+    "AUTO_RECONCILE_ON_COMPLETE": "0",
+    "CACHE_LATENTS": "0",
+    "CHECKPOINTING_STEPS": "500",
+    "CHECKPOINTS_TOTAL_LIMIT": "3",
+    "CUDA_VISIBLE_DEVICES": "0,1,2,3",
+    "DATASET_POOL_SEED": "42",
+    "DATASET_SPECS": "<local-path>",
+    "DEPTH_ENCODING": "relative_inverse_grayscale",
+    "DEPTH_PROMPT_CONTRACT": "campaign_detailed_inverse",
+    "DEPTH_SOURCE_MULTIPLIERS": "{}",
+    "DEPTH_TARGET_RESIZE_POLICY": "metric_field_then_encode",
+    "DEPTH_WEIGHT": "0.23529411764705882",
+    "DISTRIBUTED_STRATEGY": "fsdp",
+    "EVAL_GPU_GRES": "gpu:h100:4",
+    "EVAL_PARTITION": "h100",
+    "EVAL_QOS": "normal",
+    "EXCLUDE_DEPTH_SOURCE": "",
+    "EXCLUDE_NORMALS_SOURCE": "",
+    "FSDP_TRANSFORMER_DTYPE_POLICY": "fp32_all",
+    "GRADIENT_ACCUMULATION_STEPS": "2",
+    "GUIDANCE_SCALE": "3.5",
+    "IDENTITY_ARCHIVE": "<local-path>",
+    "IDENTITY_WEIGHT": "0.11764705882352941",
+    "INSTANCE_WEIGHT": "0.09411764705882353",
+    "LEARNING_RATE": "2e-6",
+    "LOSS_MASK_EROSION_RADIUS": "0",
+    "LOSS_MASK_MIN_SAMPLE_VALID_FRACTION": "0.0",
+    "LOSS_MASK_MIN_VALID_FRACTION": "0.5",
+    "LOSS_MASK_MODE": "hard",
+    "LOSS_MASK_POLICY": "auto",
+    "LOSS_MASK_REDUCTION": "per_sample",
+    "LR_SCHEDULER": "constant_with_warmup",
+    "LR_WARMUP_STEPS": "1000",
+    "MAX_TRAIN_STEPS": "37500",
+    "METRIC_DEPTH_AUX_WEIGHT": "0",
+    "MIXED_PRECISION": "bf16",
+    "MODEL": "<local-path>",
+    "MODEL_FAMILY": "flux2_klein",
+    "NORMALS_WEIGHT": "0.11764705882352941",
+    "NORMAL_PROMPT_CONTRACT": "campaign_short",
+    "NORMAL_SOURCE_MULTIPLIERS": "{}",
+    "NORMAL_TARGET_ENCODING": "vision_banana_x_left_red",
+    "NORMAL_TARGET_RESIZE_POLICY": "vector_field_renormalize_then_encode",
+    "NUM_PROCESSES": "4",
+    "OUTPUT_DIR": "<local-path>",
+    "PALETTE_MANIFOLD_LOSS_WEIGHT": "0",
+    "PD12M_WEIGHT": "0.0",
+    "PERMANENT_CHECKPOINT_STEPS": "20000,37500",
+    "PICO_ARCHIVE": "<local-path>",
+    "PICO_WEIGHT": "0.17647058823529413",
+    "PROJECT_ROOT": "<local-path>",
+    "PROMPT_ENCODING_MODE": "auto",
+    "REFSEG_WEIGHT": "0.09411764705882353",
+    "REPORT_TO": "tensorboard",
+    "RESAMPLE_DATASET_EACH_EPOCH": "1",
+    "RESOLUTION": "512",
+    "RESUME_FROM_CHECKPOINT": "latest",
+    "ROOT": "<local-path>",
+    "RUN_DIR": "<local-path>",
+    "RUN_ID": "c375_creative_no_generation",
+    "SAMPLES_PER_EPOCH": "160000",
+    "SEED": "42",
+    "SEMANTIC_WEIGHT": "0.16470588235294117",
+    "SKIP_FINAL_INFERENCE": "1",
+    "SKIP_FINAL_SAVE": "1",
+    "TASK_HOMOGENEOUS_BATCHES": "1",
+    "TRAINING_SUITE": "geometry_phase2_20260806_v1",
+    "TRAIN_ALL_TRANSFORMER_BLOCKS": "1",
+    "TRAIN_BATCH_SIZE": "1",
+    "TRAIN_FULL_TRANSFORMER": "0",
+    "TRAIN_INPUT_LAYERS": "0",
+    "TRAIN_LAST_SINGLE_TRANSFORMER_BLOCKS": "0",
+    "TRAIN_LAST_TRANSFORMER_BLOCKS": "0",
+    "TRAIN_METHOD": "partial_full_rank",
+    "TRAIN_NON_BLOCK_LAYERS": "1",
+    "TRAIN_OUTPUT_LAYERS": "1",
+    "UNIT_NORMAL_LOSS_WEIGHT": "0",
+    "WEIGHTING_SCHEME": "none",
+    "WEIGHT_DECAY": "1e-4"
+  },
+  "evaluate_checkpoints": [
+    37500
+  ],
+  "evaluation_scope": "geometry_and_compact_guardrails",
+  "evaluation_suites": [
+    "geometry",
+    "semantic",
+    "refseg",
+    "instance",
+    "imgedit",
+    "genai"
+  ],
+  "group": "creative",
+  "run_id": "c375_creative_no_generation",
+  "schema_version": 1,
+  "source_campaign_id": "geometry_phase2_20260806_v1",
+  "source_run_id": "p2_creative_no_generation_s42",
+  "status": "PLANNED"
+}

ablation_models/additional_studies/cross_task/creative_task_removal/no_image_generation/checkpoint-37500/training_args.json

@@ -0,0 +1,129 @@
+{
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "adam_epsilon": 1e-08,
+  "adam_weight_decay": 0.0001,
+  "adam_weight_decay_text_encoder": 0.001,
+  "allow_large_prompt_precompute_cache": false,
+  "allow_legacy_hypersim_range_depth": false,
+  "allow_tf32": false,
+  "allow_unverified_normal_sources": false,
+  "angular_normal_aux_weight": 0.0,
+  "aspect_ratio_buckets": "512,512;576,448;448,576;608,416;416,608",
+  "bnb_quantization_config_path": null,
+  "cache_dir": null,
+  "cache_latents": false,
+  "caption_column": "text",
+  "center_crop": false,
+  "checkpointing_steps": 500,
+  "checkpoints_total_limit": 3,
+  "class_data_dir": null,
+  "cond_image_column": "conditioning_image",
+  "dataloader_num_workers": 0,
+  "dataset_config_name": null,
+  "dataset_index_num_workers": 16,
+  "dataset_name": null,
+  "dataset_pool_seed": 42,
+  "dataset_specs": "<local-path>",
+  "depth_prompt_contract": "campaign_detailed_inverse",
+  "depth_target_encoding": "relative_inverse_grayscale",
+  "depth_target_resize_policy": "metric_field_then_encode",
+  "do_fp8_training": false,
+  "enable_npu_flash_attention": false,
+  "final_validation_prompt": null,
+  "fsdp_text_encoder": false,
+  "fsdp_transformer": true,
+  "fsdp_transformer_cls_to_wrap": "Flux2TransformerBlock,Flux2SingleTransformerBlock",
+  "fsdp_transformer_dtype_policy": "fp32_all",
+  "gradient_accumulation_steps": 2,
+  "gradient_checkpointing": true,
+  "guidance_scale": 3.5,
+  "hub_model_id": null,
+  "hub_token": null,
+  "image_column": "image",
+  "instance_data_dir": null,
+  "instance_prompt": "[TASK: IMAGE_EDITING] Edit the image according to the instruction.",
+  "learning_rate": 2e-06,
+  "local_rank": 0,
+  "logging_dir": "logs",
+  "logit_mean": 0.0,
+  "logit_std": 1.0,
+  "lora_alpha": 64,
+  "lora_dropout": 0.0,
+  "lora_layers": null,
+  "lora_rank": 64,
+  "loss_mask_erosion_radius": 0,
+  "loss_mask_min_sample_valid_fraction": 0.0,
+  "loss_mask_min_valid_fraction": 0.5,
+  "loss_mask_mode": "hard",
+  "loss_mask_policy": "auto",
+  "loss_mask_reduction": "per_sample",
+  "lr_num_cycles": 1,
+  "lr_power": 1.0,
+  "lr_scheduler": "constant_with_warmup",
+  "lr_warmup_steps": 1000,
+  "max_grad_norm": 1.0,
+  "max_sequence_length": 512,
+  "max_train_steps": 37500,
+  "metric_depth_aux_max_depth": 80.0,
+  "metric_depth_aux_weight": 0.0,
+  "mixed_precision": "bf16",
+  "mode_scale": 1.29,
+  "normal_prompt_contract": "campaign_short",
+  "normal_target_encoding": "vision_banana_x_left_red",
+  "normal_target_resize_policy": "vector_field_renormalize_then_encode",
+  "num_train_epochs": 1,
+  "num_validation_images": 4,
+  "offload": false,
+  "optimizer": "AdamW",
+  "output_dir": "<local-path>",
+  "palette_manifold_loss_weight": 0.0,
+  "permanent_checkpoint_steps": "20000,37500",
+  "precompute_cache_dir": null,
+  "pretrained_model_name_or_path": "<local-path>",
+  "prodigy_beta3": null,
+  "prodigy_decouple": true,
+  "prodigy_safeguard_warmup": true,
+  "prodigy_use_bias_correction": true,
+  "prompt_embedding_batch_size": 16,
+  "prompt_encoding_mode": "auto",
+  "prompt_precompute_max_unique": 2048,
+  "push_to_hub": false,
+  "random_flip": false,
+  "repeats": 1,
+  "report_to": "tensorboard",
+  "resample_dataset_each_epoch": true,
+  "resolution": 512,
+  "resume_from_checkpoint": "latest",
+  "revision": null,
+  "sample_batch_size": 4,
+  "samples_per_epoch": 160000,
+  "scale_lr": false,
+  "seed": 42,
+  "skip_final_inference": true,
+  "skip_final_save": true,
+  "task_homogeneous_batches": true,
+  "train_all_transformer_blocks": true,
+  "train_batch_size": 1,
+  "train_full_transformer": false,
+  "train_input_layers": false,
+  "train_last_single_transformer_blocks": 0,
+  "train_last_transformer_blocks": 0,
+  "train_method": "partial_full_rank",
+  "train_non_block_layers": true,
+  "train_output_layers": true,
+  "train_single_transformer_block_indices": null,
+  "train_transformer_block_indices": null,
+  "unit_normal_loss_weight": 0.0,
+  "upcast_before_saving": false,
+  "use_8bit_adam": false,
+  "validation_epochs": 50,
+  "validation_height": null,
+  "validation_image": null,
+  "validation_num_inference_steps": 28,
+  "validation_prompt": null,
+  "validation_prompts_file": null,
+  "validation_width": null,
+  "variant": null,
+  "weighting_scheme": "none"
+}