transformers.js/tests
Joshua Lochner c2c45cb577
Improve support of conversational models (#658)
* Add `return_full_text` option for text-generation models

* [wip] Support chat inputs in text-generation pipeline

* Align return type with python version

* Remove conversational task (moved to text-generation)

* Fix typos
2024-04-11 00:58:50 +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 Gemma Tokenizer (#598) 2024-02-21 16:22:22 +02:00
generation.test.js Improve support of conversational models (#658) 2024-04-11 00:58:50 +02:00
hub.test.js Fix failing tests 2023-05-05 00:05:03 +02:00
init.js Remove GH-specific redirect logic (#630) 2024-03-07 02:47:32 +02:00
maths.test.js Fix channel format when padding non-square images for certain models (#655) 2024-03-20 14:35:30 +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 Fix channel format when padding non-square images for certain models (#655) 2024-03-20 14:35:30 +02:00
requirements.txt Add support for RoFormer models (#464) 2023-12-26 15:49:11 +02:00
tensor.test.js Fix channel format when padding non-square images for certain models (#655) 2024-03-20 14:35:30 +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 Update tokenizer `apply_chat_template` functionality (#647) 2024-03-20 15:22:01 +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