From 4396350271ef24f361ac10966f3213bc8e8db322 Mon Sep 17 00:00:00 2001
From: Isotr0py <2037008807@qq.com>
Date: Sat, 4 Mar 2023 13:59:22 +0800
Subject: [PATCH] Add network_args to meta if using another network

---
 train_network.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train_network.py b/train_network.py
index 7aee6514..ba2f4464 100644
--- a/train_network.py
+++ b/train_network.py
@@ -427,9 +427,9 @@ def train(args):
         "ss_bucket_info": json.dumps(dataset.bucket_info),
     }
 
-  # uncomment if another network is added
-  # for key, value in net_kwargs.items():
-  #   metadata["ss_arg_" + key] = value
+  if args.network_args:
+    for key, value in net_kwargs.items():
+      metadata["ss_arg_" + key] = value
 
   if args.pretrained_model_name_or_path is not None:
     sd_model_name = args.pretrained_model_name_or_path