Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "cdf19c501de82aefb922e3718d3def753a3ba4bb"
Unverified Commit c29bec48 authored by Ceyda Cinarel's avatar Ceyda Cinarel Committed by GitHub
Browse files

fix parameter name in docstring (#21032)

parent 61e068e5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/pipelines/feature_extraction.py
View file @ c29bec48
...@@ -44,7 +44,7 @@ class FeatureExtractionPipeline(Pipeline): ...@@ -44,7 +44,7 @@ class FeatureExtractionPipeline(Pipeline):
If no framework is specified, will default to the one currently installed. If no framework is specified and If no framework is specified, will default to the one currently installed. If no framework is specified and
both frameworks are installed, will default to the framework of the `model`, or to PyTorch if no model is both frameworks are installed, will default to the framework of the `model`, or to PyTorch if no model is
provided. provided.
return_tensor (`bool`, *optional*): return_tensors (`bool`, *optional*):
If `True`, returns a tensor according to the specified framework, otherwise returns a list. If `True`, returns a tensor according to the specified framework, otherwise returns a list.
task (`str`, defaults to `""`): task (`str`, defaults to `""`):
A task-identifier for the pipeline. A task-identifier for the pipeline.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment