utils.py 115 Bytes
Newer Older
chenpangpang's avatar
chenpangpang committed
1
2
3
4
5
import torch.nn.functional as F


def is_torch2_available():
    return hasattr(F, "scaled_dot_product_attention")