/Uni-Fold

Replace attention in Uni-Fold with attention in DeepSpeed

Uni-Fold

The original Uni-Fold