search

LeapLabTHU / Agent-Attention

Official repository of Agent Attention (ECCV2024)
473 stars 35 forks source link

Agent-attention #24

Closed john09282922 closed 7 months ago

john09282922 commented 7 months ago

Hi, I would like to use your model into DDPM or DDIM method. Is it possible? Could you tell me which code or file is really important to add to them?

tian-qing001 commented 7 months ago

Hi @john09282922. Agent attention is a plug-in module that can be applied to various Transformer models. We provide example code here for applying AgentSD to PLMS and DDIM samplers.