diff --git a/launch.py b/launch.py
index f45b88f..fea858a 100644
--- a/launch.py
+++ b/launch.py
@@ -3,7 +3,7 @@ import sys
 import platform
 
 from modules.launch_util import commit_hash, fooocus_tag, is_installed, run, python, \
-    run_pip, repo_dir, git_clone, requirements_met
+    run_pip, repo_dir, git_clone, requirements_met, script_path, dir_repos
 
 
 REINSTALL_ALL = False
@@ -27,6 +27,7 @@ def prepare_environment():
     print(f"Commit hash: {commit}")
 
     git_clone(comfy_repo, repo_dir('StabilityAI-official-comfyui'), "Inference Engine", comfy_commit_hash)
+    sys.path.append(os.path.join(script_path, dir_repos, 'StabilityAI-official-comfyui'))
 
     if REINSTALL_ALL or not is_installed("torch") or not is_installed("torchvision"):
         run(f'"{python}" -m {torch_command}', "Installing torch and torchvision", "Couldn't install torch", live=True)
diff --git a/requirements_versions.txt b/requirements_versions.txt
index f530802..75d9cf0 100644
--- a/requirements_versions.txt
+++ b/requirements_versions.txt
@@ -9,4 +9,6 @@ scipy==1.9.3
 tqdm==4.64.1
 psutil==5.9.5
 opencv-python==4.7.0.72
-numpy==1.23
+numpy==1.23.5
+pytorch_lightning==1.9.4
+omegaconf==2.2.3
diff --git a/webui.py b/webui.py
index 1976c55..f76a3ef 100644
--- a/webui.py
+++ b/webui.py
@@ -1 +1,5 @@
 print('hello word')
+
+from nodes import CheckpointLoader
+
+print(CheckpointLoader)