Aitrepreneur's picture
Upload 128 files
96fc080 verified
Raw
History Blame Contribute Delete
115 Bytes
import torch.nn.functional as F
def is_torch2_available():
return hasattr(F, "scaled_dot_product_attention")