{
  "H": 192,
  "W": 384,
  "z_ch": 8,
  "dim": 768,
  "depth": 12,
  "heads": 12,
  "patch": 3,
  "action_dim": 6,
  "seq_len": 8,
  "ctx": 7,
  "vae_base": 64,
  "k_steps": 8
}