Skip to content

Commit e6cde7b

Browse files
committed
Tune Anima TeaCache coefficients
Signed-off-by: akshatvishu <akshatnayak197@gmail.com>
1 parent 431003f commit e6cde7b

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

vllm_omni/diffusion/cache/teacache/config.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -75,13 +75,14 @@
7575
],
7676
# LongCat Image transformer coefficients
7777
"LongCatImageTransformer2DModel": [652.5980, -424.1615, 84.5526, -4.5923, 0.1694],
78-
# Anima transformer coefficients (copied from Qwen-Image, tune specifically in future)
78+
# Anima transformer coefficients estimated from 70 prompts with branch-aware
79+
# CFG collection.
7980
"AnimaTransformer3DModel": [
80-
-4.50000000e02,
81-
2.80000000e02,
82-
-4.50000000e01,
83-
3.20000000e00,
84-
-2.00000000e-02,
81+
-100.26122373358824,
82+
46.465363415631714,
83+
2.3366374688249225,
84+
-0.6316431317883843,
85+
0.038771589719745,
8586
],
8687
}
8788

0 commit comments

Comments
 (0)