| # 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. | |