From c11c14590aa5d34985fd12a0b1100d4e51343ae9 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Fri, 5 Dec 2025 08:08:11 -0800
Subject: [PATCH] Image: Better LLM variation default prompt

---
 modules/shared.py              | 2 +-
 modules/ui_image_generation.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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}"