Pranav

Fix attention layers map for SD-2-1-Base

2

The [safetensors](https://huggingface.co/stabilityai/stable-diffusion-2-1-base/tree/main/vae?show_file_info=vae%2Fdiffusion_pytorch_model.safetensors) are not mapped correctly for attention layers of SD-2-1-base.

Stable diffusion mlx

15

Sharded stable diffusion inference for mlx #159 Changes: - Sharded Stable Diffusion 2-1 Base mlx - Handled diffusion steps by looping the whole model - Added back inference state -...