1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Upgrading to latest gradio release

pull/630/head
bmaltais 2023-04-17 20:54:55 -04:00
parent f1934f264a
commit 2e07329088
4 changed files with 27 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kohya_gui.py
View File

@ -20,7 +20,7 @@ def UI(**kwargs):
print('Load CSS...') print('Load CSS...')
css += file.read() + '\n' css += file.read() + '\n'
interface = gr.Blocks(css=css, title='Kohya_ss GUI') interface = gr.Blocks(css=css, title='Kohya_ss GUI', theme=gr.themes.Default())
with interface: with interface:
with gr.Tab('Dreambooth'): with gr.Tab('Dreambooth'):

33
lora_gui.py
View File

@ -870,8 +870,8 @@ def lora_tab(
value=1, value=1,
step=0.1, step=0.1,
interactive=True, interactive=True,
info='alpha for LoRA weight scaling',
) )
with gr.Row(visible=False) as LoCon_row: with gr.Row(visible=False) as LoCon_row:
# 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) # 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)
@ -926,6 +926,7 @@ def lora_tab(
label='Max resolution', label='Max resolution',
value='512,512', value='512,512',
placeholder='512,512', placeholder='512,512',
info='The maximum resolution of dataset images. W,H',
) )
stop_text_encoder_training = gr.Slider( stop_text_encoder_training = gr.Slider(
minimum=0, minimum=0,
@ -933,8 +934,10 @@ def lora_tab(
value=0, value=0,
step=1, step=1,
label='Stop text encoder training', label='Stop text encoder training',
info='After what % of steps should the text encoder stop being trained. 0 = train for all steps.',
) )
enable_bucket = gr.Checkbox(label='Enable buckets', value=True) enable_bucket = gr.Checkbox(label='Enable buckets', value=True,
info='Allow non similar resolution dataset images to be trained on.',)
with gr.Accordion('Advanced Configuration', open=False): with gr.Accordion('Advanced Configuration', open=False):
with gr.Row(visible=True) as kohya_advanced_lora: with gr.Row(visible=True) as kohya_advanced_lora:
@ -942,39 +945,47 @@ def lora_tab(
with gr.Row(visible=True): with gr.Row(visible=True):
down_lr_weight = gr.Textbox( down_lr_weight = gr.Textbox(
label='Down LR weights', label='Down LR weights',
placeholder='(Optional) eg: 0,0,0,0,0,0,1,1,1,1,1,1' placeholder='(Optional) eg: 0,0,0,0,0,0,1,1,1,1,1,1',
info='Specify the learning rate weight of the down blocks of U-Net.'
) )
mid_lr_weight = gr.Textbox( mid_lr_weight = gr.Textbox(
label='Mid LR weights', label='Mid LR weights',
placeholder='(Optional) eg: 0.5' placeholder='(Optional) eg: 0.5',
info='Specify the learning rate weight of the mid block of U-Net.'
) )
up_lr_weight = gr.Textbox( up_lr_weight = gr.Textbox(
label='Up LR weights', label='Up LR weights',
placeholder='(Optional) eg: 0,0,0,0,0,0,1,1,1,1,1,1' placeholder='(Optional) eg: 0,0,0,0,0,0,1,1,1,1,1,1',
info='Specify the learning rate weight of the up blocks of U-Net. The same as down_lr_weight.'
) )
block_lr_zero_threshold = gr.Textbox( block_lr_zero_threshold = gr.Textbox(
label='Blocks LR zero threshold', label='Blocks LR zero threshold',
placeholder='(Optional) eg: 0.1' placeholder='(Optional) eg: 0.1',
info='If the weight is not more than this value, the LoRA module is not created. The default is 0.'
) )
with gr.Tab(label='Blocks'): with gr.Tab(label='Blocks'):
with gr.Row(visible=True): with gr.Row(visible=True):
block_dims = gr.Textbox( block_dims = gr.Textbox(
label='Block dims', label='Block dims',
placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2' placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2',
info='Specify the dim (rank) of each block. Specify 25 numbers.'
) )
block_alphas = gr.Textbox( block_alphas = gr.Textbox(
label='Block alphas', label='Block alphas',
placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2' placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2',
info='Specify the alpha of each block. Specify 25 numbers as with block_dims. If omitted, the value of network_alpha is used.'
) )
with gr.Tab(label='Conv'): with gr.Tab(label='Conv'):
with gr.Row(visible=True): with gr.Row(visible=True):
conv_dims = gr.Textbox( conv_dims = gr.Textbox(
label='Conv dims', label='Conv dims',
placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2' placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2',
info='Expand LoRA to Conv2d 3x3 and specify the dim (rank) of each block. Specify 25 numbers.'
) )
conv_alphas = gr.Textbox( conv_alphas = gr.Textbox(
label='Conv alphas', label='Conv alphas',
placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2' placeholder='(Optional) eg: 2,2,2,2,4,4,4,4,6,6,6,6,8,6,6,6,6,4,4,4,4,2,2,2,2',
info='Specify the alpha of each block when expanding LoRA to Conv2d 3x3. Specify 25 numbers. If omitted, the value of conv_alpha is used.'
) )
with gr.Row(): with gr.Row():
no_token_padding = gr.Checkbox( no_token_padding = gr.Checkbox(
@ -984,7 +995,7 @@ def lora_tab(
label='Gradient accumulate steps', value='1' label='Gradient accumulate steps', value='1'
) )
weighted_captions = gr.Checkbox( weighted_captions = gr.Checkbox(
label='Weighted captions', value=False label='Weighted captions', value=False, info='Enable weighted captions in the standard style (token:1.3). No commas inside parens, or shuffle/dropout may break the decoder.',
) )
with gr.Row(): with gr.Row():
prior_loss_weight = gr.Number( prior_loss_weight = gr.Number(

3
requirements.txt
View File

@ -7,7 +7,8 @@ diffusers[torch]==0.10.2
easygui==0.98.3 easygui==0.98.3
einops==0.6.0 einops==0.6.0
ftfy==6.1.1 ftfy==6.1.1
gradio==3.19.1; sys_platform != 'darwin' gradio==3.27.0; sys_platform != 'darwin'
# gradio==3.19.1; sys_platform != 'darwin'
gradio==3.23.0; sys_platform == 'darwin' gradio==3.23.0; sys_platform == 'darwin'
lion-pytorch==0.0.6 lion-pytorch==0.0.6
opencv-python==4.7.0.68 opencv-python==4.7.0.68

2
tools/gradio_theme_builder.py Normal file
View File

@ -0,0 +1,2 @@
import gradio as gr
gr.themes.builder()