From 8ac15961e7c6ef047d59f588e00e795507d8bb0a Mon Sep 17 00:00:00 2001 From: fiatrete Date: Fri, 16 Jun 2023 15:58:57 +0800 Subject: [PATCH] optimize stable_diffusion function --- example_modules/demo_modules/stable_diffusion.module.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/example_modules/demo_modules/stable_diffusion.module.py b/example_modules/demo_modules/stable_diffusion.module.py index 2a76864..78c3e08 100644 --- a/example_modules/demo_modules/stable_diffusion.module.py +++ b/example_modules/demo_modules/stable_diffusion.module.py @@ -196,7 +196,9 @@ NOTE: Just reply using these information, don't ask me anything. if params is None: # TODO: Say something to comfort our users here? # await context.reply_text("") - params = await get_default_sd_params(original_prompt) + # Seems that only require English in system prompt does not work in new GPT version, + # GPT will reply in the language of the input, thus emphasize it in our prompt, to make it return English + params = await get_default_sd_params(original_prompt + "(You MUST reply in English)") else: params = params.copy() params.update({'prompt': original_prompt + ", " + params['prompt']})