diff --git a/modules/image/text_to_image/ernie_vilg/module.py b/modules/image/text_to_image/ernie_vilg/module.py
index 7fc56390601f99ae4dfa648999496a32808c8b89..5cc9987c950e31ec6d42e8f0a82eb71b9090f28c 100644
--- a/modules/image/text_to_image/ernie_vilg/module.py
+++ b/modules/image/text_to_image/ernie_vilg/module.py
@@ -37,6 +37,7 @@ class ErnieVilG:
 
         :param text_prompts: Phrase, sentence, or string of words and phrases describing what the image should look like.
         :param style: Image stype, currently supported 油画、水彩、粉笔画、卡通、儿童画、蜡笔画
+        :param topk: Top k images to save.
         :output_dir: Output directory
         """
         if not os.path.exists(output_dir):