transformers
add return_tensor parameter for feature extraction
#19257

Merged

add return_tensor parameter for feature extraction #19257

sgugger merged 4 commits into huggingface:main from ajsanjoaquin:feature-extraction-return-tensor

LysandreJik requested a review from

Narsil 3 years ago

Narsil commented on 2022-10-05

ajsanjoaquin force pushed 3 years ago

Narsil commented on 2022-10-06

ajsanjoaquin force pushed 3 years ago

add return_tensors parameter for feature_extraction w/ test

4f9922f1

call parameter directly

4af10377

Fixup.

da1c7be4

Narsil force pushed to da1c7be4 3 years ago

Narsil approved these changes on 2022-10-17

sgugger approved these changes on 2022-10-17

Update src/transformers/pipelines/feature_extraction.py

efa30d09

sgugger merged 35bd089a into main 3 years ago

ajsanjoaquin deleted the feature-extraction-return-tensor branch 3 years ago

Reviewers

sgugger

Narsil

Assignees

No one assigned

Labels

None yet

Milestone

No milestone