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

write a random seed to metadata

Browse Source
This commit is contained in:
ddPn08
2023-03-30 16:49:03 +09:00
parent 45381b188c
commit a7d302e196
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
train_network.py
View File

@@ -57,8 +57,9 @@ def train(args):
use_dreambooth_method = args.in_json is None use_dreambooth_method = args.in_json is None
use_user_config = args.dataset_config is not None use_user_config = args.dataset_config is not None
if args.seed is not None: if args.seed is None:
set_seed(args.seed) args.seed = random.randint(0, 2**32)
set_seed(args.seed)
tokenizer = train_util.load_tokenizer(args) tokenizer = train_util.load_tokenizer(args)