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