关于自适应注意力重加权机制

[Mrginger1]

亲爱的作者您好！请问具体在代码的哪个位置实现了将z_t去噪并预测 为x_0？

[Mrginger1]

论文中是在像素空间作的ASS，但是我看您的代码，似乎是在潜空间进行ASS的，并没有用到LDM的解码器进行解码后再作ASS?

[supremePSS]

大佬您好，请问潜空间进行ASS的代码具体在哪个文件的哪一行？

[Mrginger1]

大佬您好，请问潜空间进行ASS的代码具体在哪个文件的哪一行？

您好呀，我觉得ASS实现是在ldm/models/diffusion/ddim.py 的第157行dis_map = ((new_mask * (img - x0)) ** 2).mean(dim=1)及后，img和x0这两个似乎都是潜空间的变量而非像素空间，我不确定我的理解是否有问题，你可以自己判断一下0v0

[supremePSS]

感谢！