AttributeError: module 'torch.nn.functional' has no attribute 'scaled_dot_product_attention'

NVlabs / MambaVision

Official PyTorch Implementation of MambaVision: A Hybrid Mamba-Transformer Vision Backbone

https://arxiv.org/abs/2407.08083

Other

792 stars 43 forks source link

Closed SegmGod closed 3 months ago

ahatamiz commented 3 months ago

Could you please install pip install torch == 2.1.2 ? The torch version is most likely outdated.