stable-diffusion alignment Attention-UNet
stable-diffusion based JAX implementation for layer weights.
- Input
- 1337-dim embedding
- Encoder
- 26 x Attention-UNet with 4 heads
- Output
- recall projection
Training config
optimizer=Adagrad, lr=0.328, scheduler=polynomial, warmup=1760