unet.py file miss count_flops_attn() function.

andreas128 / RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

1.88k stars 153 forks source link

unet.py file miss count_flops_attn() function. #35

Open VoldemortGin opened 1 year ago

VoldemortGin commented 1 year ago

In guided_diffusion.unet.py file, the method count_flops() of 2 classes (QKVAttentionLegacy class and QKVAttention class) need a function with signature count_flops_attn(model, _x, y). The method is not been called, so there is no error, but I think it's better to adding it.