oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-02-02 13:54:40 +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-04 10:50:40 -08:00
parent ffef3c7b1d
commit a90739f498
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

@ -320,7 +320,7 @@ settings = {
'image_batch_size': 1,
'image_batch_count': 1,
'image_llm_variations': False,
'image_llm_variations_prompt': 'Please create a creative variation of the image generation prompt above. Keep the same general subject and style, but vary the details, composition, lighting, or mood. Respond with only the new prompt, nothing else.',
'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_model_menu': 'None',
'image_dtype': 'bfloat16',
'image_attn_backend': 'sdpa',

2
modules/ui_image_generation.py
View file

@ -704,7 +704,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 = 'Please create a creative variation of the image generation prompt above. Keep the same general subject and style, but vary the details, composition, lighting, or mood. Respond with only the new prompt, nothing else.'
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.'
augmented_message = f"{prompt}\n\n=====\n\n{variation_instruction}"