From 547344321678a0a13b038fbf8d9853d59dba502c Mon Sep 17 00:00:00 2001 From: chenjian Date: Tue, 16 Aug 2022 05:04:45 +0000 Subject: [PATCH] fix --- modules/image/text_to_image/ernie_vilg/module.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/image/text_to_image/ernie_vilg/module.py b/modules/image/text_to_image/ernie_vilg/module.py index 7fc56390..5cc9987c 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): -- GitLab