Duplicated from multimodalart/face-to-all
be88838
1
2
3
4
5
6
import torch.nn.functional as F def is_torch2_available(): return hasattr(F, "scaled_dot_product_attention")