diff --git a/modules/shared.py b/modules/shared.py index f6e86bdf..3f3742de 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -322,7 +322,7 @@ settings = { 'image_batch_size': 1, 'image_batch_count': 1, 'image_llm_variations': False, - 'image_llm_variations_prompt': 'Write a variation of the image generation prompt above. Consider the intent of the user with that prompt and write something that will likely please them, with added details. Your reply should contain the new prompt and nothing else.', + 'image_llm_variations_prompt': 'Write a variation of the image generation prompt above. Consider the intent of the user with that prompt and write something that will likely please them, with added details. Output only the new prompt. Do not add any explanations, prefixes, or additional text.', 'image_model_menu': 'None', 'image_dtype': 'bfloat16', 'image_attn_backend': 'sdpa', diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py index 28d4c627..c55d2438 100644 --- a/modules/ui_image_generation.py +++ b/modules/ui_image_generation.py @@ -701,7 +701,7 @@ def generate_prompt_variation(state): # Get the custom variation prompt or use default variation_instruction = state.get('image_llm_variations_prompt', '') if not variation_instruction: - variation_instruction = 'Write a variation of the image generation prompt above. Consider the intent of the user with that prompt and write something that will likely please them, with added details. Your reply should contain the new prompt and nothing else.' + variation_instruction = 'Write a variation of the image generation prompt above. Consider the intent of the user with that prompt and write something that will likely please them, with added details. Output only the new prompt. Do not add any explanations, prefixes, or additional text.' augmented_message = f"{prompt}\n\n=====\n\n{variation_instruction}"