File size: 712 Bytes
bdb1898 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | # Prompt Reinjection Rotations
Released Procrustes rotation files for the ICML 2026 paper
`Prompt Reinjection: Alleviating Prompt Forgetting in Multimodal Diffusion Transformers`.
## Files
- `sd3_coco5k_o1.pt`: SD3 rotation computed on COCO 5k, with origin layer 1 and target layers 2-23.
- `flux_coco5k_o2.pt`: FLUX rotation computed on COCO 5k, with origin layer 2 and target layers 3-56.
## Usage
Download the files and place them under `prompt_reinjection/rotations/` in the main repository:
```bash
hf download LewisYao/PromptReinjection sd3_coco5k_o1.pt flux_coco5k_o2.pt --local-dir prompt_reinjection/rotations
```
The default helper script in the main repository will pick them up automatically.
|