diff --git a/src/transformers/pipelines/feature_extraction.py b/src/transformers/pipelines/feature_extraction.py index e8adb11b687da6..7d67a615ac02d2 100644 --- a/src/transformers/pipelines/feature_extraction.py +++ b/src/transformers/pipelines/feature_extraction.py @@ -24,7 +24,7 @@ class FeatureExtractionPipeline(Pipeline): >>> extractor = pipeline(model="google-bert/bert-base-uncased", task="feature-extraction") >>> result = extractor("This is a simple test.", return_tensors=True) - >>> result.shape # This is a tensor of shape [1, sequence_lenth, hidden_dimension] representing the input string. + >>> result.shape # This is a tensor of shape [1, sequence_length, hidden_dimension] representing the input string. torch.Size([1, 8, 768]) ```