[FEAT] Use flash-attn in ViT instead of torch.sdpa
#396
The logs for this run have expired and are no longer available.
Loading
flash-attn in ViT instead of torch.sdpa
#396