Joshua Lochner cdcbfc125c Add support for Segment Anything Model (#510) ...

* Update SamModel

* Make `AutoModel.from_pretrained` work with SamModel

* Add listed support for SAM (Segment Anything Model)

* Update types of `calculateDimensions`

* Throw error if reading image from tensor with dims.length != 3

* Make SamProcessor input points optional

* Fix type errors

* `let` -> `const`

* `cat` -> `stack`

* Expose `reshape_input_points` in `SamProcessor`

* Add `input_labels` input parameter for SAM

* Add `input_labels` to sam processor

* Update SAM unit tests

* Remove TODOs

* Update JSDoc

2024-01-10 17:47:21 +02:00

..

extra

Add support for cross-encoder models (+fix token type ids) (#501)

2024-01-04 18:37:26 +02:00

convert.py

Add phi-1_5 model (#493)

2024-01-02 23:12:09 +02:00

requirements.txt

Use transformers==4.33.2 in conversion script

2023-09-23 17:06:27 +02:00

supported_models.py

Add support for Segment Anything Model (#510)

2024-01-10 17:47:21 +02:00