From a37c0171a5494ab637a9336b2ff7eb9bad4d952c Mon Sep 17 00:00:00 2001 From: Jingyi Date: Fri, 17 Nov 2023 21:03:31 +0800 Subject: [PATCH] resort wait tip --- source/panel/image_generation.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/source/panel/image_generation.py b/source/panel/image_generation.py index 69a160d9..d4eb68a2 100644 --- a/source/panel/image_generation.py +++ b/source/panel/image_generation.py @@ -116,12 +116,16 @@ def get_llm_processed_prompts(initial_prompt): ) conversation.prompt = sd_prompt + """ Example: [Positive Prompt: visually appealing, high-quality image of a big, large, muscular horse with powerful body, majestic stance, flowing mane, detailed texture, vivid color, striking photography., Negative Prompt: ugly, distorted, inappropriate or NSFW content, Recommended Model List: ["sd_xl_base_1.0.safetensors"]] """ + + st.write("Wait for LLM to process the prompt...") + response = conversation.predict(input=initial_prompt) logger.info("the first invoke: {}".format(response)) # logger.info("the second invoke: {}".format(conversation.predict(input="change to realist style"))) @@ -428,7 +432,7 @@ def generate_llm_image(initial_prompt: str, llm_prompt: bool = True): negative = "" models = default_models if llm_prompt is True: - st.write("Wait for LLM to process the prompt...") + positive_prompt, negative_prompt, model_list = get_llm_processed_prompts(prompt) # if prompt is empty, use default