From 8eac99599ad9a645f608d3814c63a0297e877de8 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 4 Dec 2025 19:58:06 -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 7b572dec..f6e86bdf 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': 'Your task is to create a creative variation of the image generation prompt above. Keep the main subject but feel free to add an interesting setting, scenario, pose, atmosphere, or stylistic twist. Be specific and vivid. Respond with only the new prompt, 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. Your reply should contain the new prompt and nothing else.',
     '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 2cad5dc4..139fd891 100644
--- a/modules/ui_image_generation.py
+++ b/modules/ui_image_generation.py
@@ -694,7 +694,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 = 'Your task is to create a creative variation of the image generation prompt above. Keep the main subject but feel free to add an interesting setting, scenario, pose, atmosphere, or stylistic twist. Be specific and vivid. Respond with only the new prompt, 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. Your reply should contain the new prompt and nothing else.'
 
     augmented_message = f"{prompt}\n\n=====\n\n{variation_instruction}"