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 d2a4083ae5704716e33351aed03af8f3cc45f3ae.

* 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>
This commit is contained in:
Armaghan Shakir
2025-03-31 14:45:07 +05:00
committed by GitHub
parent f99c279d20
commit 0710e9b1e8
12 changed files with 612 additions and 7 deletions

View File

@@ -149,12 +149,24 @@ alt="drawing" width="900"/>
[[autodoc]] SamImageProcessor
## SamVisionModel
[[autodoc]] SamVisionModel
- forward
## SamModel
[[autodoc]] SamModel
- forward
## TFSamVisionModel
[[autodoc]] TFSamVisionModel
- call
## TFSamModel
[[autodoc]] TFSamModel