Hugging Face's logo

LewisYao
/
PromptReinjection

Model card Files
xet
 Community
PromptReinjection / README.md
LewisYao's picture
LewisYao
Add model card for released Prompt Reinjection rotations
bdb1898 verified
preview code
|
raw
history blame contribute delete
712 Bytes

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:

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.