mlx-examples/stable_diffusion/stable_diffusion
Pawel Kowalski fc1495abaa
Stable diffusion - check model weights shape and support int for "attention_head_dim" (#85)
* Allow integer as attention_head_dim
* Reshape downloaded weights to match model if there is a mismatch
2023-12-15 13:01:02 -08:00
..
__init__.py format 2023-11-30 11:52:47 -08:00
clip.py add copyright in source 2023-11-30 11:08:53 -08:00
config.py add copyright in source 2023-11-30 11:08:53 -08:00
model_io.py Stable diffusion - check model weights shape and support int for "attention_head_dim" (#85) 2023-12-15 13:01:02 -08:00
sampler.py format 2023-11-30 11:52:47 -08:00
tokenizer.py hf correction 2023-12-12 17:08:04 -08:00
unet.py add copyright in source 2023-11-30 11:08:53 -08:00
vae.py fix: typo in variable name 2023-12-07 21:30:04 +00:00