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