diff --git a/nodes.py b/nodes.py
index 4450d29..b0ea745 100644
--- a/nodes.py
+++ b/nodes.py
@@ -388,7 +388,7 @@ class InitFluxLoRATraining:
             "max_data_loader_n_workers": 0,
             "seed": 42,
             "gradient_checkpointing": True,
-            "network_module": "networks.lora_flux",
+            "network_module": ".networks.lora_flux",
             "dataset_config": dataset_toml,
             "output_name": f"{output_name}_rank{kwargs.get('network_dim')}_{save_dtype}",
             "loss_type": "l2",
diff --git a/train_network.py b/train_network.py
index 5f7f3ef..61ae8d2 100644
--- a/train_network.py
+++ b/train_network.py
@@ -352,9 +352,10 @@ class NetworkTrainer:
         pbar.update(1)
 
         # Load the model for incremental learning
-        sys.path.append(os.path.dirname(__file__))
+        #sys.path.append(os.path.dirname(__file__))
         accelerator.print("import network module:", args.network_module)
-        network_module = importlib.import_module(args.network_module)
+        package = __name__.split('.')[0]
+        network_module = importlib.import_module(args.network_module, package=package)
 
         if args.base_weights is not None:
             # base_weights が指定されている場合は、指定された重みを読み込みマージする