File size: 115 Bytes
96fc080 |
1 2 3 4 5 6 |
import torch.nn.functional as F
def is_torch2_available():
return hasattr(F, "scaled_dot_product_attention")
|
96fc080 |
1 2 3 4 5 6 |
import torch.nn.functional as F
def is_torch2_available():
return hasattr(F, "scaled_dot_product_attention")
|