Skip to content

Commit

Permalink
Cleanup and bump localization to 23.0.11
Browse files Browse the repository at this point in the history
  • Loading branch information
hinablue committed Mar 18, 2024
1 parent 81b3d81 commit ffb6430
Show file tree
Hide file tree
Showing 10 changed files with 501 additions and 545 deletions.
2 changes: 1 addition & 1 deletion kohya_gui/blip_caption_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def list_train_dirs(path):
with gr.Row():
caption_file_ext = gr.Textbox(
label="Caption file extension",
placeholder="Extension for caption file, e.g., .caption, .txt",
placeholder="Extension for caption file (e.g., .caption, .txt)",
value=".txt",
interactive=True,
)
Expand Down
10 changes: 5 additions & 5 deletions kohya_gui/class_advanced_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@ def full_options_update(full_fp16, full_bf16):
],
value='75',
)

with gr.Row():
if training_type == "lora":
self.fp8_base = gr.Checkbox(
Expand All @@ -144,7 +144,7 @@ def full_options_update(full_fp16, full_bf16):
value=False,
info='Required bitsandbytes >= 0.36.0',
)

self.full_fp16.change(
full_options_update,
inputs=[self.full_fp16, self.full_bf16],
Expand Down Expand Up @@ -249,7 +249,7 @@ def full_options_update(full_fp16, full_bf16):
minimum=0,
maximum=1,
step=0.01,
info='recommended values are 0.05 - 0.15',
info='Recommended values are 0.05 - 0.15',
)
self.adaptive_noise_scale = gr.Slider(
label='Adaptive noise scale',
Expand All @@ -266,15 +266,15 @@ def full_options_update(full_fp16, full_bf16):
minimum=0,
maximum=64,
step=1,
info='enable multires noise (recommended values are 6-10)',
info='Enable multires noise (recommended values are 6-10)',
)
self.multires_noise_discount = gr.Slider(
label='Multires noise discount',
value=0,
minimum=0,
maximum=1,
step=0.01,
info='recommended values are 0.8. For LoRAs with small datasets, 0.1-0.3',
info='Recommended values are 0.8. For LoRAs with small datasets, 0.1-0.3',
)
self.noise_offset_type.change(
noise_offset_type_change,
Expand Down
4 changes: 2 additions & 2 deletions kohya_gui/extract_lycoris_locon_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ def extract_lycoris_locon(

# Run the command
subprocess.run(run_cmd, shell=True, env=env)

log.info('Done extracting...')


Expand Down Expand Up @@ -166,7 +166,7 @@ def list_save_to(path):
current_save_dir = path
return list(list_files(path, exts=[".safetensors"], all=True))

with gr.Tab("Extract LyCORIS LoCON"):
with gr.Tab("Extract LyCORIS LoCon"):
gr.Markdown(
"This utility can extract a LyCORIS LoCon network from a finetuned model."
)
Expand Down
6 changes: 3 additions & 3 deletions kohya_gui/git_caption_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,19 +104,19 @@ def list_train_dirs(path):
with gr.Row():
caption_ext = gr.Textbox(
label='Caption file extension',
placeholder='Extention for caption file. eg: .caption, .txt',
placeholder='Extension for caption file (e.g., .caption, .txt)',
value='.txt',
interactive=True,
)

prefix = gr.Textbox(
label='Prefix to add to BLIP caption',
label='Prefix to add to GIT caption',
placeholder='(Optional)',
interactive=True,
)

postfix = gr.Textbox(
label='Postfix to add to BLIP caption',
label='Postfix to add to GIT caption',
placeholder='(Optional)',
interactive=True,
)
Expand Down
10 changes: 5 additions & 5 deletions kohya_gui/lora_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -844,8 +844,8 @@ def train_model(
return
text_encoder_lr_float = float(text_encoder_lr)
unet_lr_float = float(unet_lr)



# Determine the training configuration based on learning rate values
if text_encoder_lr_float == 0 and unet_lr_float == 0:
Expand Down Expand Up @@ -1035,7 +1035,7 @@ def lora_tab(

def list_presets(path):
json_files = []

# Insert an empty string at the beginning
json_files.insert(0, "none")

Expand Down Expand Up @@ -1133,7 +1133,7 @@ def list_presets(path):
minimum=0,
maximum=1,
)

unet_lr = gr.Number(
label="Unet learning rate",
value="0.0001",
Expand Down Expand Up @@ -1199,7 +1199,7 @@ def list_presets(path):
decompose_both = gr.Checkbox(
value=False,
label="LoKr decompose both",
info=" Controls whether both input and output dimensions of the layer's weights are decomposed into smaller matrices for reparameterization.",
info="Controls whether both input and output dimensions of the layer's weights are decomposed into smaller matrices for reparameterization.",
visible=False,
)
train_on_input = gr.Checkbox(
Expand Down
2 changes: 1 addition & 1 deletion kohya_gui/manual_caption_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -303,7 +303,7 @@ def list_images_dirs(path):
load_images_button = gr.Button('Load', elem_id='open_folder')
caption_ext = gr.Textbox(
label='Caption file extension',
placeholder='Extension for caption file. eg: .caption, .txt',
placeholder='Extension for caption file (e.g., .caption, .txt)',
value='.txt',
interactive=True,
)
Expand Down
2 changes: 1 addition & 1 deletion kohya_gui/merge_lycoris_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -210,7 +210,7 @@ def list_save_to(path):
)

with gr.Row():
is_sdxl = gr.Checkbox(label='is sdxl', value=False, interactive=True)
is_sdxl = gr.Checkbox(label='is SDXL', value=False, interactive=True)
is_v2 = gr.Checkbox(label='is v2', value=False, interactive=True)

merge_button = gr.Button('Merge model')
Expand Down
2 changes: 1 addition & 1 deletion kohya_gui/resize_lora_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ def list_save_to(path):
)
with gr.Row():

verbose = gr.Checkbox(label='Verbose', value=True)
verbose = gr.Checkbox(label='Verbose logging', value=True)
save_precision = gr.Radio(
label='Save precision',
choices=['fp16', 'bf16', 'float'],
Expand Down
4 changes: 2 additions & 2 deletions kohya_gui/wd14_caption_gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,7 @@ def list_train_dirs(path):

caption_extension = gr.Textbox(
label='Caption file extension',
placeholder='Extention for caption file. eg: .caption, .txt',
placeholder='Extension for caption file (e.g., .caption, .txt)',
value='.txt',
interactive=True,
)
Expand Down Expand Up @@ -230,7 +230,7 @@ def list_train_dirs(path):
character_threshold = gr.Slider(
value=0.35,
label='Character threshold',
info='useful if you want to train with character',
info='Useful if you want to train with character',
minimum=0,
maximum=1,
step=0.05,
Expand Down
Loading

0 comments on commit ffb6430

Please sign in to comment.