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.
state_dict
ckpt