Support for SDPA for SAM models #34110
ArthurZucker
merged
19 commits into
huggingface:main
from
MagnusS0:sam_support_sdpa
ArthurZucker
commented
on 2024-10-24
ArthurZucker
approved these changes
on 2024-10-30
feat: add support for sdpa and gradient checkpointing
6dd779f8
fix: ruff format
fcf3bcb9
fix: config sdpa
40637ea1
fix: sdpa layer naming convention
ee086179
fix: update test_eager_matches_sdpa_inference to handle vision_hidden…
e93fcfa1
test: skip incompatible tests and fix loading issue with sdpa
70fe9bc7
style: apply Ruff formatting
7ee64dc4
ruff fix again after rebase
df4e50c2
Merge branch 'huggingface:main' into sam_support_sdpa
a2bff23d
[run-slow] sam
c6ea29a7
Merge branch 'main' into sam_support_sdpa
3a356b0d
Merge branch 'main' into sam_support_sdpa
0ed4b69c
[run-slow] sam
69560e42
zucchini-nlp
commented
on 2024-11-19
refactor: Address review comments and improve sub-config handling in …
4ef96b3e
[run-slow] sam
b401e946
style: ruff
b4707461
[run-slow] sam
e12c1880
style: ruff again ...
dbd416c8
[run-slow] sam
d572d198
ArthurZucker
commented
on 2024-12-17
Reviewers
Assignees
No one assigned
Labels
Vision
SDPA
run-slow
Milestone
No milestone
Login to write a write a comment.
Login via GitHub