StableDiffusion/KohyaSS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #291 from bmaltais/dev

Browse Source 
Fix for LoCon sliders visibility
This commit is contained in:
bmaltais 2023-03-03 21:58:08 -05:00 committed by GitHub
commit e562356b98
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
lora_gui.py
View File

@ -252,9 +252,9 @@ def open_configuration(
# This next section is about making the LoCon parameters visible if LoRA_type = 'Standard' # This next section is about making the LoCon parameters visible if LoRA_type = 'Standard'
if my_data.get('LoRA_type', 'Standard') == 'LoCon': if my_data.get('LoRA_type', 'Standard') == 'LoCon':
values.append(gr.Group.update(visible=True)) values.append(gr.Row.update(visible=True))
else: else:
values.append(gr.Group.update(visible=False)) values.append(gr.Row.update(visible=False))
return tuple(values) return tuple(values)
@ -712,35 +712,34 @@ def lora_tab(
interactive=True, interactive=True,
) )
with gr.Group(visible=False) as LoCon_group: with gr.Row(visible=False) as LoCon_row:
def LoRA_type_change(LoRA_type): # locon= gr.Checkbox(label='Train a LoCon instead of a general LoRA (does not support v2 base models) (may not be able to some utilities now)', value=False)
if LoRA_type == 'LoCon': conv_dim = gr.Slider(
return gr.Group.update(visible=True) minimum=1,
else: maximum=512,
return gr.Group.update(visible=False) value=1,
step=1,
with gr.Row(): label='LoCon Convolution Rank (Dimension)',
# locon= gr.Checkbox(label='Train a LoCon instead of a general LoRA (does not support v2 base models) (may not be able to some utilities now)', value=False)
conv_dim = gr.Slider(
minimum=1,
maximum=512,
value=1,
step=1,
label='LoCon Convolution Rank (Dimension)',
)
conv_alpha = gr.Slider(
minimum=1,
maximum=512,
value=1,
step=1,
label='LoCon Convolution Alpha',
)
# Show of hide LoCon conv settings depending on LoRA type selection
LoRA_type.change(
LoRA_type_change, inputs=[LoRA_type], outputs=[LoCon_group]
) )
conv_alpha = gr.Slider(
minimum=1,
maximum=512,
value=1,
step=1,
label='LoCon Convolution Alpha',
)
# Show of hide LoCon conv settings depending on LoRA type selection
def LoRA_type_change(LoRA_type):
print('LoRA type changed...')
if LoRA_type == 'LoCon':
return gr.Group.update(visible=True)
else:
return gr.Group.update(visible=False)
LoRA_type.change(
LoRA_type_change, inputs=[LoRA_type], outputs=[LoCon_row]
)
with gr.Row(): with gr.Row():
max_resolution = gr.Textbox( max_resolution = gr.Textbox(
label='Max resolution', label='Max resolution',
@ -910,7 +909,7 @@ def lora_tab(
button_open_config.click( button_open_config.click(
open_configuration, open_configuration,
inputs=[config_file_name] + settings_list, inputs=[config_file_name] + settings_list,
outputs=[config_file_name] + settings_list + [LoCon_group], outputs=[config_file_name] + settings_list + [LoCon_row],
) )
button_save_config.click( button_save_config.click(