v0xie / sd-webui-agentattention

Speed up image generation and improve image quality using Agent Attention.
GNU General Public License v3.0
39 stars 3 forks source link

Fix: Restore attn1/attn2 in remove_patch, add recommended remove_patch statement in callback #4

Closed v0xie closed 6 months ago

v0xie commented 6 months ago

Partially addresses #3

Previously, remove_patch() did not restore the class value of attn1/attn2. We store the old value and restore in remove_patch.