diff --git a/lightning_logs/version_0/events.out.tfevents.1676227718.DESKTOP-4ESQENU.2348.0 b/lightning_logs/version_0/events.out.tfevents.1676227718.DESKTOP-4ESQENU.2348.0
new file mode 100644
index 0000000..99da6cc
Binary files /dev/null and b/lightning_logs/version_0/events.out.tfevents.1676227718.DESKTOP-4ESQENU.2348.0 differ
diff --git a/lightning_logs/version_0/hparams.yaml b/lightning_logs/version_0/hparams.yaml
new file mode 100644
index 0000000..0967ef4
--- /dev/null
+++ b/lightning_logs/version_0/hparams.yaml
@@ -0,0 +1 @@
+{}
diff --git a/tool_add_control.py b/tool_add_control.py
index 6bd832a..8076b51 100644
--- a/tool_add_control.py
+++ b/tool_add_control.py
@@ -11,6 +11,7 @@ assert not os.path.exists(output_path), 'Output filename already exists.'
 assert os.path.exists(os.path.dirname(output_path)), 'Output path is not valid.'
 
 import torch
+from share import *
 from cldm.model import create_model