mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2025-12-06 07:12:10 +01:00
Lint
This commit is contained in:
parent
0c4bdb19c8
commit
b70011aeea
|
|
@ -365,7 +365,6 @@ def apply_image_model_cli_overrides():
|
||||||
settings['image_quant'] = args.image_quant
|
settings['image_quant'] = args.image_quant
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def fix_loader_name(name):
|
def fix_loader_name(name):
|
||||||
if not name:
|
if not name:
|
||||||
return name
|
return name
|
||||||
|
|
|
||||||
|
|
@ -538,7 +538,6 @@ def create_event_handlers():
|
||||||
generate, gradio('interface_state'), gradio('image_output_gallery'), show_progress=False).then(
|
generate, gradio('interface_state'), gradio('image_output_gallery'), show_progress=False).then(
|
||||||
lambda: [gr.update(visible=False), gr.update(visible=True)], None, gradio('image_generating_btn', 'image_generate_btn'))
|
lambda: [gr.update(visible=False), gr.update(visible=True)], None, gradio('image_generating_btn', 'image_generate_btn'))
|
||||||
|
|
||||||
|
|
||||||
# Model management
|
# Model management
|
||||||
shared.gradio['image_refresh_models'].click(
|
shared.gradio['image_refresh_models'].click(
|
||||||
lambda: gr.update(choices=utils.get_available_image_models()),
|
lambda: gr.update(choices=utils.get_available_image_models()),
|
||||||
|
|
@ -713,7 +712,7 @@ def generate(state):
|
||||||
|
|
||||||
total_images = int(state['image_batch_count']) * int(state['image_batch_size'])
|
total_images = int(state['image_batch_count']) * int(state['image_batch_size'])
|
||||||
total_steps = state["image_steps"] * int(state['image_batch_count'])
|
total_steps = state["image_steps"] * int(state['image_batch_count'])
|
||||||
logger.info(f'Generated {total_images} images in {(t1-t0):.2f} seconds ({total_steps/(t1-t0):.2f} steps/s, seed {seed})')
|
logger.info(f'Generated {total_images} images in {(t1 - t0):.2f} seconds ({total_steps / (t1 - t0):.2f} steps/s, seed {seed})')
|
||||||
|
|
||||||
return all_images
|
return all_images
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue