guided-diffusion icon indicating copy to clipboard operation
guided-diffusion copied to clipboard

`AttentionBlock` forces `use_checkpoint` to be `True`, is it a feature or a bug?

Open baofff opened this issue 2 years ago • 0 comments

In Line 297 of unet.py, I find the flag argument of checkpoint(...) is set to True. This is unlike other modules, which set the flag argument to self.use_checkpoint. Is it a feature or a bug?

baofff avatar Aug 16 '22 10:08 baofff