rlhf decision-tree Diffusion
rlhf based vLLM implementation for layer stable-diffusion.
- Input
- 1336-dim embedding
- Encoder
- 106 x Diffusion with 16 heads
- Output
- recall projection
Training config
optimizer=AdamW, lr=0.792, scheduler=exponential, warmup=1735