transformers/tests/models/sam
Armaghan Shakir 0710e9b1e8
Create and Expose SamVisionModel as public for better accessibility (#36493)
* move encoder below

* auto modeling

* write SamVisionTester

* fix vision attention shape

* fix SamVisionTest

* minor changes to SamVisionTest

* Revert "fix vision attention shape"

This reverts commit d2a4083ae5.

* fix attention output shape in new tests

* remove encoder examples

* run modular on got_ocr2

* code formatting

* fix got_ocr2

* ruff fixes

* code quality

* add sam_vision in auto modeling and auto configuration

* remove composite test

* updated index.md

* add TFSamVisionEncoder to __init__

* fix public TFSamVisionEncoder

* remove outdated todo comment

* set test_torch_exportable

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* rename: VisionEncoder -> VisionModel

* bring back original SamVisionEncoder

* rename back: VisionEncoderOutput -> VisionModelOutput

* undo changes in SamModelTester

* reuse SamVisionEncoder in SamVisionModel

---------

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
2025-03-31 11:45:07 +02:00
..
__init__.py Add Segment Anything Model (SAM) (#22654) 2023-04-19 21:01:49 +02:00
test_modeling_sam.py Create and Expose SamVisionModel as public for better accessibility (#36493) 2025-03-31 11:45:07 +02:00
test_modeling_tf_sam.py Create and Expose SamVisionModel as public for better accessibility (#36493) 2025-03-31 11:45:07 +02:00
test_processor_sam.py 🚨🚨🚨 Fix sdpa in SAM and refactor relative position embeddings (#36422) 2025-03-17 09:39:52 +00:00