mode: gradient
bsz: 1
iters: 8
direction: ArBr
max_length: 1024
dtype: fp32
scale: stable
stable_gamma: 16