VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-10 15:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train_network.py

Browse Source
This commit is contained in:
gesen2egee
2024-10-07 18:01:00 +08:00
parent 012e7e63a5
commit f09824fd31
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
train_network.py
View File

@@ -205,6 +205,9 @@ class NetworkTrainer:
ds_for_collator = train_dataset_group if args.max_data_loader_n_workers == 0 else None ds_for_collator = train_dataset_group if args.max_data_loader_n_workers == 0 else None
collator = train_util.collator_class(current_epoch, current_step, ds_for_collator) collator = train_util.collator_class(current_epoch, current_step, ds_for_collator)
if args.no_token_padding:
train_dataset_group.disable_token_padding()
if args.debug_dataset: if args.debug_dataset:
train_util.debug_dataset(train_dataset_group) train_util.debug_dataset(train_dataset_group)
return return
@@ -1162,6 +1165,11 @@ def setup_parser() -> argparse.ArgumentParser:
nargs="*", nargs="*",
help="additional arguments for network (key=value) / ネットワークへの追加の引数", help="additional arguments for network (key=value) / ネットワークへの追加の引数",
) )
parser.add_argument(
"--no_token_padding",
action="store_true",
help="disable token padding (same as Diffuser's DreamBooth) / トークンのpaddingを無効にする（Diffusers版DreamBoothと同じ動作）",
)
parser.add_argument( parser.add_argument(
"--network_train_unet_only", action="store_true", help="only training U-Net part / U-Net関連部分のみ学習する" "--network_train_unet_only", action="store_true", help="only training U-Net part / U-Net関連部分のみ学習する"
) )