Magnus 6eb00dd2f0 Support for SDPA for SAM models (#34110) ...

* feat: add support for sdpa and gradient checkpointing

* fix: ruff format

* fix: config sdpa

* fix: sdpa layer naming convention

* fix: update test_eager_matches_sdpa_inference to handle vision_hidden_states

* test: skip incompatible tests and fix loading issue with sdpa

- Updated tests to skip cases flash and dynamic compile.
- Minor adjustment to ensure correct loading of model with sdpa for dispatch test.

* style: apply Ruff formatting

* ruff fix again after rebase

* [run-slow] sam

* [run-slow] sam

* refactor: Address review comments and improve sub-config handling in SAM model tests

- Added attributes for sub_configs as per PR #34410.
- Enabled tests for configs, ensuring the composite model (SAM) has several sub-configs in the main config.
- Added class attribute _is_composite=True to the tester class
- test_sdpa_can_dispatch_composite_models added

* [run-slow] sam

* style: ruff

* [run-slow] sam

* style: ruff again ...

* [run-slow] sam

2024-12-17 14:46:05 +01:00

..

__init__.py

Add Segment Anything Model (SAM) (#22654)

2023-04-19 21:01:49 +02:00

test_modeling_sam.py

Support for SDPA for SAM models (#34110)

2024-12-17 14:46:05 +01:00

test_modeling_tf_sam.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_processor_sam.py

Remove repeated prepare_images in processor tests (#33163)

2024-09-09 13:20:27 +01:00