Init with Zelda and MC, only finetune DiT

Ckpt step = 20k