sageattention assert headdim in [64, 96, 128], "headdim should be in [64, 96, 128]."

Open zhaoyun0071 opened this issue 10 months ago • 1 comments

1、when use sageattention in 14b_i2V get error bellow

sageattention\core.py", line 82, in sageattn assert headdim in [64, 96, 128], "headdim should be in [64, 96, 128]."

2、but in 14b_t2V ，sageattention works

Mar 04 '25 13:03 zhaoyun0071

@zhaoyun0071 Thanks for reporting this issue. We will debug and fix it.

Mar 04 '25 13:03 Artiprocher

@zhaoyun0071 Thanks for reporting this issue. We will debug and fix it.

hi, have you solved this bug ?

Mar 10 '25 02:03 lith0613

@zhaoyun0071 @lith0613 We disable the sage attention in the image encoder of I2V model. This bug is fixed.

Mar 10 '25 10:03 Artiprocher