transformers
Add PyTorch version check for FA backend on AMD GPUs
#35813

Merged

Add PyTorch version check for FA backend on AMD GPUs #35813

SunMarc merged 1 commit into huggingface:main from mht-sharma:sdpa_rocm_version

Disable FA backend for SDPA on AMD GPUs (PyTorch < 2.4.1)

958dd3f4

mht-sharma requested a review from

SunMarc 1 year ago

SunMarc approved these changes on 2025-01-22

SunMarc merged fdcc62c8 into main 1 year ago

Reviewers

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone