LewisYao commited on
Commit
fa82be2
·
verified ·
1 Parent(s): c9a5158

Add PromptReinjection weights and model card

Browse files
Files changed (3) hide show
  1. README.md +19 -0
  2. flux_coco5k_o2.pt +3 -0
  3. sd3_coco5k_o1.pt +3 -0
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Prompt Reinjection Rotations
2
+
3
+ Released Procrustes rotation files for the ICML 2026 paper
4
+ `Prompt Reinjection: Alleviating Prompt Forgetting in Multimodal Diffusion Transformers`.
5
+
6
+ ## Files
7
+
8
+ - `sd3_coco5k_o1.pt`: SD3 rotation computed on COCO 5k, with origin layer 1 and target layers 2-23.
9
+ - `flux_coco5k_o2.pt`: FLUX rotation computed on COCO 5k, with origin layer 2 and target layers 3-56.
10
+
11
+ ## Usage
12
+
13
+ Download the files and place them under `prompt_reinjection/rotations/` in the main repository:
14
+
15
+ ```bash
16
+ hf download LewisYao/PromptReinjection sd3_coco5k_o1.pt flux_coco5k_o2.pt --local-dir prompt_reinjection/rotations
17
+ ```
18
+
19
+ The default helper script in the main repository will pick them up automatically.
flux_coco5k_o2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac898a6b15e76d2a428ed7664afb0813fe1c17840eb665b32f6f4f9289ccc21f
3
+ size 2038433542
sd3_coco5k_o1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b47cede1c2d5c373627de21d57542ef5fd9e781fbf91f32b6b95822d8431463
3
+ size 207619663