Use Flash Attention for CLIP image encoder

#52

by YenChunChen - opened Jul 16, 2024

←

Jul 16, 2024

•

As title, FA2 for training speedup.

@haipingwu please review

YenChunChen changed pull request status to open Jul 16, 2024

Microsoft org Jul 17, 2024

hi @leoxiaobin , please merge this PR

leoxiaobin changed pull request status to merged Jul 17, 2024

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment