From d4883cb50f270e1a79deb1cef0b0b07ac321fcd4 Mon Sep 17 00:00:00 2001
From: riyadhctg <3258642+riyadhctg@users.noreply.github.com>
Date: Wed, 28 Sep 2022 12:07:52 +0800
Subject: [PATCH] add param def for multi_code

---
 modeling_utils.py | 3 +++
 1 file changed, 3 insertions(+)

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