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

Small fixes

Browse source
This commit is contained in:
oobabooga 2025-11-27 15:48:53 -08:00
parent 21f992e7f7
commit 666816a773
1 changed files with 65 additions and 66 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
modules/ui_image_generation.py
View file

@ -337,7 +337,6 @@ def generate(prompt, neg_prompt, width, height, steps, seed, batch_size_parallel
model_menu, dtype_dropdown, attn_dropdown, cpu_offload_checkbox, compile_checkbox): model_menu, dtype_dropdown, attn_dropdown, cpu_offload_checkbox, compile_checkbox):
"""Generate images with the current model settings.""" """Generate images with the current model settings."""
# Get current UI values (these are Gradio components, we need their values)
model_name = shared.image_model_name model_name = shared.image_model_name
if model_name == 'None': if model_name == 'None':
@ -345,15 +344,15 @@ def generate(prompt, neg_prompt, width, height, steps, seed, batch_size_parallel
# Auto-load model if not loaded # Auto-load model if not loaded
if shared.image_model is None: if shared.image_model is None:
# Load saved settings for the model # Get effective settings (CLI > yaml > defaults)
saved_settings = load_image_model_settings() settings = get_effective_settings()
result = load_image_model( result = load_image_model(
model_name, model_name,
dtype=saved_settings['dtype'], dtype=settings['dtype'],
attn_backend=saved_settings['attn_backend'], attn_backend=settings['attn_backend'],
cpu_offload=saved_settings['cpu_offload'], cpu_offload=settings['cpu_offload'],
compile_model=saved_settings['compile_model'] compile_model=settings['compile_model']
) )
if result is None: if result is None: