Skip to content

Commit

Permalink
Update modeling_utils.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Cyrilvallez committed Dec 17, 2024
1 parent 0d3d3e3 commit 49135d0
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/transformers/modeling_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -1545,8 +1545,6 @@ def _autoset_attn_implementation(
"Using the `SDPA` attention implementation on multi-gpu setup with ROCM may lead to performance issues due to the FA backend. Disabling it to use alternative backends."
)
torch.backends.cuda.enable_flash_sdp(False)
elif config._attn_implementation in ALL_ATTENTION_FUNCTIONS:
pass
elif isinstance(requested_attn_implementation, dict):
config._attn_implementation = None
else:
Expand Down

0 comments on commit 49135d0

Please sign in to comment.