Attempted to implement a comfyui node.

[JettHu]

I tried to implement a version of comfyui's T-GATE but seemed to encounter some problems. The quality of the generated image declined after the apply the node.

this is the repo: https://github.com/JettHu/ComfyUI_TGate

[WentianZhang-ML]

Hi, thanks for your attention to our work.

I have gone through your repo. Please do not use TGATE on "attn1" and try again.

"attn1" refers to the self-attention when only_cross_attention is set as False (default config), but TGATE is only used to cache the cross-attention output

[JettHu]

Hi, thanks for your attention to our work.

I have gone through your repo. Please do not use TGATE on "attn1" and try again.

"attn1" refers to the self-attention when only_cross_attention is set as False (default config), but TGATE is only used to cache the cross-attention output

thx, i'll try it

[JettHu]

It works after I add a only_cross_attention switch!

[WentianZhang-ML]

Sure, you can also set only_cross_attention as False and only use TGATE on attn2, which is closer to our pipeline.

[JettHu]

I'm going to close this issue