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.