unet.mid_block.attentions.0_k10_hidden640_auxk256_bs4096_lr0.0001/final/config.json

{"n_dirs_local": 640, "d_model": 1280, "k": 10, "auxk": 256, "dead_steps_threshold": 2441}