{
  "base_model": "/lustre/fs1/portfolios/nvr/projects/nvr_elm_llm/users/hcai/workspace/code/.cache/huggingface/hub/models--black-forest-labs--FLUX.2-klein-base-4B/snapshots/a3b4f4849157f664bdbc776fd7453c2783562f4d",
  "checkpoint": "/lustre/fs1/portfolios/nvr/projects/nvr_elm_llm/users/hcai/workspace/code/junyiwu/FAR-Anyflow1.5/experiments/flux2_klein_base_4b_diffusionnft_alphagrpo_dvreward_alphagrpo20k_512px_20step_cfg4_16prompts_group14_7train_1dvreward_tp1_2node_nrt/models/step_500.pt",
  "checkpoint_sha256": "2fc9365453f4ff19ce28493d83361dcb94bbe90b9474eabd1a3c65a3a13eb00c",
  "diffusers_version": "0.39.0",
  "dtype": "bfloat16",
  "format": "diffusers-flux2-klein-pipeline",
  "global_step": 500,
  "lora_alpha": 64.0,
  "lora_merge": {
    "adapter_names": [
      "default"
    ],
    "changed_element_count": 2447174844,
    "changed_tensor_count": 60,
    "lora_pair_count": 60,
    "max_abs_delta": 0.0116424560546875
  },
  "lora_rank": 32,
  "native_pipeline_class": "diffusers.Flux2KleinPipeline",
  "native_transformer_class": "diffusers.Flux2Transformer2DModel",
  "parameters": 3875544576,
  "scheduler": {
    "num_train_timesteps": 1000,
    "stochastic_sampling": false,
    "use_dynamic_shifting": true
  },
  "source_experiment": "flux2_klein_base_4b_diffusionnft_alphagrpo_dvreward_alphagrpo20k_512px_20step_cfg4_16prompts_group14_7train_1dvreward_tp1_2node_nrt",
  "source_revision": "a3b4f4849157f664bdbc776fd7453c2783562f4d",
  "tensor_keys": 169,
  "torch_version": "2.9.1+cu128",
  "transformer_delta_from_base": {
    "changed_element_count": 2447174844,
    "changed_tensor_count": 60,
    "changed_tensor_examples": [
      "single_transformer_blocks.0.attn.to_out.weight",
      "single_transformer_blocks.0.attn.to_qkv_mlp_proj.weight",
      "single_transformer_blocks.1.attn.to_out.weight",
      "single_transformer_blocks.1.attn.to_qkv_mlp_proj.weight",
      "single_transformer_blocks.10.attn.to_out.weight",
      "single_transformer_blocks.10.attn.to_qkv_mlp_proj.weight",
      "single_transformer_blocks.11.attn.to_out.weight",
      "single_transformer_blocks.11.attn.to_qkv_mlp_proj.weight"
    ],
    "l2_delta": 32.30985084978022,
    "max_abs_delta": 0.0116424560546875
  },
  "weight": "ema"
}