{
 "model_dir": "/workspace/models/MOSS-Image-SFT",
 "lodestones_dir": "/workspace/models/MOSS-Image-SFT-8-steps-dmd",
 "cache_dir": "/workspace/cache",
 "prompts": "/workspace/prompts/train.jsonl",
 "eval_prompts": "/workspace/prompts/eval.jsonl",
 "out_dir": "/workspace/runs/dmad",
 "steps": 4,
 "gen_rank": 128,
 "gen_alpha": 128,
 "fake_rank": 64,
 "fake_alpha": 64,
 "disc_taps": [
  5,
  11,
  17,
  23,
  29,
  35
 ],
 "disc_hidden": 256,
 "iterations": 2500,
 "micro": 2,
 "fake_per_gen": 2,
 "fake_warmup": 200,
 "lr_gen": 2e-05,
 "lr_time": 2e-05,
 "lr_q": 2e-05,
 "lr_fake": 0.0001,
 "lr_disc": 5e-05,
 "betas": [
  0.0,
  0.99
 ],
 "ema": 0.998,
 "cfg": [
  3.0,
  3.0
 ],
 "skip_neg": [
  22
 ],
 "tau_beta": [
  2.5,
  1.0
 ],
 "tau_clamp": [
  0.03,
  0.985
 ],
 "dt": 0.015625,
 "adv_weight": 0.5,
 "adv_start": 50,
 "adv_warmup": 200,
 "disc_real": "adm",
 "teacher_dir": [
  "/workspace/teacher",
  "/workspace/teacher2"
 ],
 "p_gan": 0.5,
 "gan_t": [
  0.02,
  0.98
 ],
 "clip_gen": 0.0,
 "buckets": [
  [
   1024,
   1024,
   0.35
  ],
  [
   896,
   1152,
   0.15
  ],
  [
   832,
   1216,
   0.15
  ],
  [
   1152,
   896,
   0.1
  ],
  [
   1216,
   832,
   0.1
  ],
  [
   768,
   1344,
   0.075
  ],
  [
   1344,
   768,
   0.075
  ]
 ],
 "method": "dmad",
 "critic_rank": 128,
 "critic_alpha": 128,
 "critic_start": 21,
 "critic_hidden": 512,
 "critic_shift": 5.0,
 "critic_sigma": [
  0.02,
  0.98
 ],
 "lr_critic": 4e-05,
 "wd_critic": 0.01,
 "critic_warmup": 50,
 "ckpt": true,
 "seed": 0,
 "log_every": 10,
 "sample_every": 100,
 "save_every": 200,
 "export_every": 500,
 "eval_count": 8,
 "eval_seed": 1234
}