This is a diffusers version of "nai" suitable for use with Stable Diffusion finetuning applications with the empty nested `state_dict` removed which was causing performance issues in `ckpt` format.