stable-diffusion environment Attention-UNet
stable-diffusion based TensorFlow implementation for embedding instruct.
- Input
- 5138-dim embedding
- Encoder
- 59 x Attention-UNet with 32 heads
- Output
- bleu projection
Training config
optimizer=Adam, lr=0.443, scheduler=cosine, warmup=1262