AttributeError: module 'torch.nn.functional' has no attribute 'scaled_dot_product_attention'. Did you mean: '_scaled_dot_product_attention'?

[yfpeng1234]

Hi, Dear author. I install the package as readme suggested and change pytorch version to: conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia Because my server is 3090 and driver is a little bit old. But I encountered the following error: AttributeError: module 'torch.nn.functional' has no attribute 'scaled_dot_product_attention'. Did you mean: '_scaled_dot_product_attention'? Do you know how to fix it or which verison of pytorch is supported in cambrian. Thanks!