guided-diffusion
guided-diffusion copied to clipboard
`AttentionBlock` forces `use_checkpoint` to be `True`, is it a feature or a bug?
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?