write a random seed to metadata

ertugrul-dmr · Apr 2, 2023 · a7d302e · a7d302e
1 parent 45381b1
commit a7d302e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/train_network.py b/train_network.py
@@ -57,8 +57,9 @@ def train(args):
     use_dreambooth_method = args.in_json is None
     use_user_config = args.dataset_config is not None
 
-    if args.seed is not None:
-        set_seed(args.seed)
+    if args.seed is None:
+        args.seed = random.randint(0, 2**32)
+    set_seed(args.seed)
 
     tokenizer = train_util.load_tokenizer(args)