oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Image: Better LLM variation default prompt

Browse source
This commit is contained in:
oobabooga 2025-12-05 08:08:11 -08:00
parent 0dd468245c
commit c11c14590a
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/shared.py
View file

@ -322,7 +322,7 @@ settings = {
'image_batch_size': 1, 'image_batch_size': 1,
'image_batch_count': 1, 'image_batch_count': 1,
'image_llm_variations': False, '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_model_menu': 'None',
'image_dtype': 'bfloat16', 'image_dtype': 'bfloat16',
'image_attn_backend': 'sdpa', 'image_attn_backend': 'sdpa',

2
modules/ui_image_generation.py
View file

@ -701,7 +701,7 @@ def generate_prompt_variation(state):
# Get the custom variation prompt or use default # Get the custom variation prompt or use default
variation_instruction = state.get('image_llm_variations_prompt', '') variation_instruction = state.get('image_llm_variations_prompt', '')
if not variation_instruction: 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}" augmented_message = f"{prompt}\n\n=====\n\n{variation_instruction}"