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