diff --git a/modeling_utils.py b/modeling_utils.py
index 1616b4b..304d365 100644
--- a/modeling_utils.py
+++ b/modeling_utils.py
@@ -723,6 +723,9 @@ def generate(
             num_return_sequences: (`optional`) int
                 The number of independently computed returned sequences for each element in the batch. Default to 1.
 
+            multi_code: (`optional`) list of int
+                The token for the topic of generation. Default to None.
+
         Examples::
 
             tokenizer = AutoTokenizer.from_pretrained('distilgpt2')   # Initialize tokenizer