more precise docstring

huggingface · Aug 31, 2024 · fce676b · fce676b
1 parent a5fee60
commit fce676b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/generation/configuration_utils.py b/src/transformers/generation/configuration_utils.py
@@ -289,8 +289,8 @@ class GenerationConfig(PushToHubMixin):
             more details.
         return_dict_in_generate (`bool`, *optional*, defaults to `False`):
             Whether or not to return a [`~utils.ModelOutput`], as opposed to returning exclusively the generated
-            sequence. This flag must be set to return the generation cache or other optional outputs (see flags
-            starting with `output_`)
+            sequence. This flag must be set to `True` to return the generation cache (when `use_cache` is `True`)
+            or optional outputs (see flags starting with `output_`)
 
         > Special tokens that can be used at generation time