From 68cd874bb68f30e792243659fbeb8c98733cd365 Mon Sep 17 00:00:00 2001
From: mio <74481573+mio2333@users.noreply.github.com>
Date: Fri, 10 Mar 2023 18:29:34 +0800
Subject: [PATCH] Append sys path for import_module

This will be better if we run the scripts we do not run the training script from the current directory.  This is reasonable as some other projects will use this as a subfolder, such as https://github.com/ddPn08/kohya-sd-scripts-webui. I can not run the script without adding this.
---
 train_network.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/train_network.py b/train_network.py
index cf64c894..8be8305c 100644
--- a/train_network.py
+++ b/train_network.py
@@ -134,6 +134,8 @@ def train(args):
     gc.collect()
 
   # prepare network
+  import sys
+  sys.path.append(os.path.dirname(__file__))
   print("import network module:", args.network_module)
   network_module = importlib.import_module(args.network_module)