diff --git a/src/transformers/models/plbart/modeling_plbart.py b/src/transformers/models/plbart/modeling_plbart.py index 78ba3b998bfe..99b8c09e7cb9 100644 --- a/src/transformers/models/plbart/modeling_plbart.py +++ b/src/transformers/models/plbart/modeling_plbart.py @@ -559,7 +559,7 @@ def _set_gradient_checkpointing(self, module, value=False): >>> values, predictions = probs.topk(5) >>> tokenizer.decode(predictions).split() - ['same', 'first', 'highest', 'result', 'Fib'] + ['first', 'same', 'highest', 'result', 'number'] ``` """