From 32ba7575010a3fdd024cc4058650d81988d126f1 Mon Sep 17 00:00:00 2001
From: Dalton <murraydalton1@comcast.net>
Date: Wed, 20 Mar 2024 23:55:04 -0400
Subject: [PATCH] Re-add import but after if check

---
 modules/initialize_util.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modules/initialize_util.py b/modules/initialize_util.py
index 7a661476d..8abe78814 100644
--- a/modules/initialize_util.py
+++ b/modules/initialize_util.py
@@ -27,6 +27,7 @@ def fix_torch_version():
 def fix_pytorch_lightning():
     # Checks if pytorch_lightning.utilities.distributed already exists in the sys.modules cache
     if 'pytorch_lightning.utilities.distributed' not in sys.modules:
+        import pytorch_lightning
         # Lets the user know that the library was not found and then will set it to pytorch_lightning.utilities.rank_zero
         print(f"Pytorch_lightning.distributed not found, attempting pytorch_lightning.rank_zero")
         sys.modules["pytorch_lightning.utilities.distributed"] = pytorch_lightning.utilities.rank_zero