transformers/tests
Ayrton San Joaquin 35bd089a24
add return_tensor parameter for feature extraction (#19257)
* add return_tensors parameter for feature_extraction  w/ test

add return_tensor parameter for feature extraction

Revert "Merge branch 'feature-extraction-return-tensor' of https://github.com/ajsanjoaquin/transformers into feature-extraction-return-tensor"

This reverts commit d559da743b87914e111a84a98ba6dbb70d08ad88, reversing
changes made to bbef89278650c04c090beb65637a8e9572dba222.

* call parameter directly

Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>

* Fixup.

* Update src/transformers/pipelines/feature_extraction.py

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-10-17 11:17:26 -04:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed deprecate is_torch_bf16_available (#17738) 2022-06-20 08:40:11 -04:00
extended Update self-push workflow (#17177) 2022-05-13 16:28:00 +02:00
fixtures add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629) 2022-04-11 11:06:10 +02:00
generation Add TF whisper (#19378) 2022-10-10 14:48:17 +01:00
mixed_int8 [bnb] Minor modifications (#18631) 2022-08-17 00:48:10 +02:00
models TF port of ESM (#19587) 2022-10-17 14:16:16 +01:00
onnx Add onnx support for VisionEncoderDecoder (#19254) 2022-10-10 09:20:19 -04:00
optimization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pipelines add return_tensor parameter for feature extraction (#19257) 2022-10-17 11:17:26 -04:00
repo_utils Add some tests for check_dummies (#19146) 2022-09-21 14:54:09 -04:00
sagemaker transformers-cli login => huggingface-cli login (#18490) 2022-08-06 09:42:55 +02:00
tokenization fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
trainer Run torchdynamo tests (#19056) 2022-09-15 11:10:16 -07:00
utils Image transforms library (#18520) 2022-10-12 18:32:02 +01:00
__init__.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
test_configuration_common.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_common.py Add tests for legacy load by url and fix bugs (#19078) 2022-09-16 23:20:02 +02:00
test_image_transforms.py Image transforms library (#18520) 2022-10-12 18:32:02 +01:00
test_modeling_common.py Poc to use safetensors (#19175) 2022-09-30 10:58:04 -04:00
test_modeling_flax_common.py [JAX] Replace all jax.tree_* calls with jax.tree_util.tree_* (#18361) 2022-09-09 15:18:56 +02:00
test_modeling_tf_common.py Sharding fails in TF when absolute scope was modified if . in layer name (#19124) 2022-10-14 18:34:33 +02:00
test_sequence_feature_extraction_common.py Some tests misusing assertTrue for comparisons fix (#16771) 2022-04-19 14:44:08 +02:00
test_tokenization_common.py Tokenizer from_pretrained should not use local files named like tokenizer files (#19626) 2022-10-14 14:06:56 -04:00