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!