transformers.js/tests
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
..
data Generate tests before running unit tests 2023-05-02 20:01:19 +02:00
configs.test.js Remove unnecessary imports 2023-05-31 02:06:34 +02:00
generate_tests.py Add support for cross-encoder models (+fix token type ids) (#501) 2024-01-04 18:37:26 +02:00
generation.test.js Ensure WASM fallback does not crash in GH actions (#402) 2023-11-19 08:06:49 +02:00
hub.test.js Fix failing tests 2023-05-05 00:05:03 +02:00
init.js Add support for VITS (multilingual TTS) (#466) 2023-12-26 16:34:52 +02:00
maths.test.js Add support for CLAP (`zero-shot-audio-classification`) and Audio Spectrogram Transformer (`audio-classification`) (#427) 2023-12-05 12:17:42 +02:00
models.test.js Add support for computing CLIP image and text embeddings separately (Closes #148) (#227) 2023-08-01 14:01:04 +02:00
pipelines.test.js Add support for Segformer (#480) 2023-12-27 15:12:07 +02:00
processors.test.js Add support for Segment Anything Model (#510) 2024-01-10 17:47:21 +02:00
requirements.txt Add support for RoFormer models (#464) 2023-12-26 15:49:11 +02:00
tensor.test.js Add support for chat templates (#408) 2023-12-18 17:00:50 +02:00
test_utils.js Add support for processing non-square images w/ `ConvNextFeatureExtractor` (#503) 2024-01-10 17:04:21 +02:00
tokenizers.test.js Add support for cross-encoder models (+fix token type ids) (#501) 2024-01-04 18:37:26 +02:00
utils.test.js Add support for CLAP (`zero-shot-audio-classification`) and Audio Spectrogram Transformer (`audio-classification`) (#427) 2023-12-05 12:17:42 +02:00