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

add save parser and fix lora scripts model name and hash

Browse Source
This commit is contained in:
青龍聖者@bdsqlsz
2024-02-21 19:38:12 +08:00
parent d91b1d3793
commit 86503cb945
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
stable_cascade_train_c_network.py
View File

@@ -730,8 +730,8 @@ class NetworkTrainer:
metadata["ss_network_args"] = json.dumps(net_kwargs)
# model name and hash
if args.pretrained_model_name_or_path is not None:
sd_model_name = args.pretrained_model_name_or_path
if args.stage_c_checkpoint_path is not None:
sd_model_name = args.stage_c_checkpoint_path
if os.path.exists(sd_model_name):
metadata["ss_sd_model_hash"] = train_util.model_hash(sd_model_name)
metadata["ss_new_sd_model_hash"] = train_util.calculate_sha256(sd_model_name)
@@ -992,6 +992,7 @@ def setup_parser() -> argparse.ArgumentParser:
train_util.add_tokenizer_arguments(parser)
train_util.add_dataset_arguments(parser, True, True, True)
train_util.add_training_arguments(parser, True)
train_util.add_sd_saving_arguments(parser)
train_util.add_optimizer_arguments(parser)
config_util.add_config_arguments(parser)
custom_train_functions.add_custom_train_arguments(parser)

1
stable_cascade_train_stage_c.py
View File

@@ -531,6 +531,7 @@ def setup_parser() -> argparse.ArgumentParser:
train_util.add_tokenizer_arguments(parser)
train_util.add_dataset_arguments(parser, True, True, True)
train_util.add_training_arguments(parser, False)
train_util.add_sd_saving_arguments(parser)
train_util.add_optimizer_arguments(parser)
config_util.add_config_arguments(parser)
add_sdxl_training_arguments(parser) # cache text encoder outputs